Update app.py

app.py

@@ -6,31 +6,26 @@ import numpy as np
 onnx_model_vgg19_path = "./vgg19-30epochs.onnx"
 onnx_model_inceptionv3_path = "./InceptionV3-20epochs.onnx"
 onnx_model_resnet101_path = "./Resnet101-30epochs.onnx"
-onnx_model_vgg16_path = "./vgg16.onnx"
+onnx_model_vgg16_path = "./vgg16-20epochs.onnx"
 
 
 
-class_labels = ['Ajloun Castle', 'Hadrians Arch', 'Petra-siq', 'Roman Ruins-Jerash', 'Roman amphitheater', 'The Cardo Maximus of Jerash', 'Wadi Rum', 'petra-Treasury', 'umm qais']
-class_labels2 = ['Ajloun Castle', 'Hadrians Arch', 'Petra-siq', 'petra-Treasury', 'Roman amphitheater', 'Roman Ruins-Jerash', 'The Cardo Maximus of Jerash', 'umm qais', 'Wadi Rum']
+labels = ['Ajloun Castle', 'Hadrians Arch', 'Petra-siq', 'Roman Ruins-Jerash', 'Roman amphitheater', 'The Cardo Maximus of Jerash', 'Wadi Rum', 'petra-Treasury', 'umm qais']
 
 def predict_image(image_path, model):
     
     if model == "InceptionV3":
         img_size = (550, 475)
-        labels = class_labels
         model_inceptionv3 = onnxruntime.InferenceSession(onnx_model_inceptionv3_path)
     elif model == "Resnet101":
         img_size = (250, 200)
-        labels = class_labels
         model_resnet101 = onnxruntime.InferenceSession(onnx_model_resnet101_path)
     elif model == "Vgg19":
         img_size = (250, 200)
-        labels = class_labels
         model_vgg19 = onnxruntime.InferenceSession(onnx_model_vgg19_path)
 
     elif model == "Vgg16":
-        img_size = (150, 150)
-        labels = class_labels2
+        img_size = (200, 150)
         model_vgg16 = onnxruntime.InferenceSession(onnx_model_vgg16_path)
 
     img = cv2.imread(image_path)
@@ -74,6 +69,7 @@ interface_image = gr.Interface(
     inputs=inputs_image,
     fn=predict_image,
     outputs=outputs_text,
-    title="classifier_demo"
+    title="classifier_demo",
+
 )
 interface_image.launch()