Khaled27 commited on
Commit
64cb358
1 Parent(s): 7cccfb0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -9,7 +9,7 @@ import torch
9
  app = Flask(__name__)
10
 
11
  model = AutoModelForImageClassification.from_pretrained(
12
- './myModel')
13
  image_processor = AutoImageProcessor.from_pretrained(
14
  "google/vit-base-patch16-224-in21k")
15
 
 
9
  app = Flask(__name__)
10
 
11
  model = AutoModelForImageClassification.from_pretrained(
12
+ './my_model')
13
  image_processor = AutoImageProcessor.from_pretrained(
14
  "google/vit-base-patch16-224-in21k")
15