Hugging Face's logo

Spaces:
Khaled27
/
Naptah
Runtime error

App Files Community
1
Khaled27 commited on
Commit
d142a3f
•
1 Parent(s): 64cb358

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -11,7 +11,7 @@ app = Flask(__name__)
11
  model = AutoModelForImageClassification.from_pretrained(
12
  './my_model')
13
  image_processor = AutoImageProcessor.from_pretrained(
14
- "google/vit-base-patch16-224-in21k")
15
 
16
 
17
  @app.route('/upload_image', methods=['POST'])
 
11
  model = AutoModelForImageClassification.from_pretrained(
12
  './my_model')
13
  image_processor = AutoImageProcessor.from_pretrained(
14
+ "microsoft/resnet-50")
15
 
16
 
17
  @app.route('/upload_image', methods=['POST'])