Update app.py

app.py

@@ -1,22 +1,10 @@
 import gradio as gr
 import yolov7
-import subprocess
 import tempfile
-import time
-from pathlib import Path
-import uuid
 import cv2
-import gradio as gr
-
+from pathlib import Path
 
-    
-def image_fn(
-    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,
-):
+def image_fn(image, model_path, image_size, conf_threshold, iou_threshold):
     """
     YOLOv7 inference function
     Args:
@@ -28,42 +16,36 @@ def image_fn(
     Returns:
         Rendered image
     """
-
     model = yolov7.load(model_path, device="cpu", hf_model=True, trace=False)
     model.conf = conf_threshold
     model.iou = iou_threshold
     results = model([image], size=image_size)
-    return results.render()[0]
-  
-  
-        
+    img = results.render()[0]
+    img.flags.writeable = True  # Make the image writable
+    return img
 
+# Use gr.components instead of gr.inputs and gr.outputs
 image_interface = gr.Interface(
     fn=image_fn,
     inputs=[
-    gr.inputs.Image(type="pil", label="Input Image"),
-    gr.inputs.Dropdown(
-        choices=[
-            "Aalaa/Yolov7_Visual_Pollution_Detection",
-        ],
-        default="Aalaa/Yolov7_Visual_Pollution_Detection",
-        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.outputs.Image(type="filepath", label="Output Image"),
-    
+        gr.components.Image(type="pil", label="Input Image"),
+        gr.components.Dropdown(
+            choices=["Aalaa/Yolov7_Visual_Pollution_Detection"],
+            value="Aalaa/Yolov7_Visual_Pollution_Detection",
+            label="Model"
+        ),
+        gr.components.Slider(minimum=320, maximum=1280, value=640, step=32, label="Image Size"),
+        gr.components.Slider(minimum=0.0, maximum=1.0, value=0.25, step=0.05, label="Confidence Threshold"),
+        gr.components.Slider(minimum=0.0, maximum=1.0, value=0.45, step=0.05, label="IOU Threshold")
+    ],
+    outputs=gr.components.Image(type="numpy", label="Output Image"),
     examples=[['image1.jpg', 'Aalaa/Yolov7_Visual_Pollution_Detection', 640, 0.25, 0.45]],
     cache_examples=True,
-    theme='huggingface',
+    theme='default'
 )
 
-
-
 if __name__ == "__main__":
     gr.TabbedInterface(
         [image_interface],
         ["Run on Images"],
-    ).launch()
+    ).launch()