Update app.py

app.py

@@ -1,3 +1,18 @@
+import gradio as gr
+import cv2
+import onnxruntime
+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-20epochs.onnx"
+
+
+
+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)
@@ -43,4 +58,32 @@ def predict_image(image_path, model):
         return "Add another image, confidence is below 50%"
 
     predicted_label = labels[predicted_class]
-    return predicted_label
+    return predicted_label
+
+inputs_image = [
+    gr.Image(type="filepath", label="Input Image"),
+    gr.Radio(choices=[
+        "InceptionV3",
+        "Vgg19",
+        "Vgg16",
+        "Resnet101",
+    ], type="value", label="Select_model")
+]
+outputs_text = [gr.Textbox(label="Site")]
+interface_image = gr.Interface(
+    inputs=inputs_image,
+    fn=predict_image,
+    outputs=outputs_text,
+    title="classifier_demo"
+    # examples=[["Wadi Rum.jpg", "Wadi Rum","Vgg19" ], 
+    #                   ["Ajloun Castle.jpg", "Ajloun Castle","Vgg19"],
+    #                   ["Hadrians Arch.jpg", "Hadrians Arch","Vgg19" ], 
+    #                   ["Patra_Siq.jpg", "Patra_Siq","Vgg19"],
+    #                   ["petra-Treasury.jpg", "Petra-Treasury","Vgg19" ], 
+    #                   ["Roman amphitheater.jpg",  "Vgg19","Roman amphitheater","Vgg19"],
+    #                   ["Roman Ruins-Jerash.jpg",  "Roman Ruins-Jerash","Vgg19"], 
+    #                   ["The Cardo Maximus of Jerash.jpg", "The Cardo Maximus of Jerash","Vgg19"],
+    #                   ["umm qais.jpg",  "umm qais","Vgg19"]]
+
+)
+interface_image.launch()