Update app.py

app.py

@@ -18,11 +18,11 @@ import gradio as gr
 
     
 def image_fn(
-    image: gr.Image = None,
-    model_path: gr.Dropdown = None,
-    image_size: gr.Slider = 640,
-    conf_threshold: gr.Slider = 0.25,
-    iou_threshold: gr.Slider = 0.45,
+    image: gr.inputs.Image = None,
+    model_path: gr.inputs.Dropdown = None,
+    image_size: gr.inputs.Slider = 640,
+    conf_threshold: gr.inputs.Slider = 0.25,
+    iou_threshold: gr.inputs.Slider = 0.45,
 ):
     """
     YOLOv7 inference function
@@ -93,21 +93,22 @@ def video_fn(model_path, video_file, conf_thres, iou_thres, start_sec, duration)
 image_interface = gr.Interface(
     fn=image_fn,
     inputs=[
-    gr.Image(type="pil", label="Input Image"),
-    gr.Dropdown(
+    gr.inputs.Image(type="pil", label="Input Image"),
+    gr.inputs.Dropdown(
         choices=[
-            "Aalaa/Yolov7_Visual_Pollution_Detection",
+            "alshimaa/SEE_model_yolo7",
             #"kadirnar/yolov7-v0.1",
         ],
+        default="alshimaa/SEE_model_yolo7",
         label="Model",
     )
     #gr.inputs.Slider(minimum=320, maximum=1280, default=640, step=32, label="Image Size")
     #gr.inputs.Slider(minimum=0.0, maximum=1.0, default=0.25, step=0.05, label="Confidence Threshold"),
     #gr.inputs.Slider(minimum=0.0, maximum=1.0, default=0.45, step=0.05, label="IOU Threshold")
 ],
-    outputs=gr.Image(type="filepath", label="Output Image"),
+    outputs=gr.outputs.Image(type="filepath", label="Output Image"),
     title="Smart Environmental Eye (SEE)",
-    examples=[['image1.jpg', 'Aalaa/Yolov7_Visual_Pollution_Detection', 640, 0.25, 0.45]],
+    examples=[['image1.jpg', 'alshimaa/SEE_model_yolo7', 640, 0.25, 0.45]],
     cache_examples=True,
     theme='huggingface',
 )
@@ -116,12 +117,13 @@ image_interface = gr.Interface(
 video_interface = gr.Interface(
     fn=video_fn,
     inputs=[
-        gr.Video(source = "upload", type = "mp4", label = "Input Video"),
-        gr.Dropdown(
+        gr.inputs.Video(source = "upload", type = "mp4", label = "Input Video"),
+        gr.inputs.Dropdown(
         choices=[
-            "Aalaa/Yolov7_Visual_Pollution_Detection",
+            "alshimaa/SEE_model_yolo7",
             #"kadirnar/yolov7-v0.1",
         ],
+        default="alshimaa/SEE_model_yolo7",
         label="Model",
     ),
     ],