Anas090 commited on
Commit
6f4fa81
·
verified ·
1 Parent(s): cafee2f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -16,11 +16,11 @@ class_labels2 = ['Ajloun Castle', 'Hadrians Arch', 'Petra-siq', 'petra-Treasury'
16
  def predict_image(image_path, model):
17
 
18
  if model == "InceptionV3":
19
- img_size = (475, 550)
20
  labels = class_labels
21
  model_inceptionv3 = onnxruntime.InferenceSession(onnx_model_inceptionv3_path)
22
  elif model == "Resnet101":
23
- img_size = (200, 250)
24
  labels = class_labels
25
  model_resnet101 = onnxruntime.InferenceSession(onnx_model_resnet101_path)
26
  elif model == "Vgg19":
 
16
  def predict_image(image_path, model):
17
 
18
  if model == "InceptionV3":
19
+ img_size = (550, 475)
20
  labels = class_labels
21
  model_inceptionv3 = onnxruntime.InferenceSession(onnx_model_inceptionv3_path)
22
  elif model == "Resnet101":
23
+ img_size = (250, 200)
24
  labels = class_labels
25
  model_resnet101 = onnxruntime.InferenceSession(onnx_model_resnet101_path)
26
  elif model == "Vgg19":