Update app.py

app.py

@@ -3,10 +3,10 @@ import torch
 from ultralyticsplus import YOLO, render_result
 
 
-def yoloV9_func(image: gr.inputs.Image = None,
-                image_size: gr.inputs.Slider = 640,
-                conf_threshold: gr.inputs.Slider = 0.4,
-                iou_threshold: gr.inputs.Slider = 0.50):
+def yoloV9_func(image: gr.Image = None,
+                image_size: int = 640,
+                conf_threshold: float = 0.4,
+                iou_threshold: float = 0.50):
     """This function performs YOLOv9 object detection on the given image.
 
     Args:
@@ -37,17 +37,17 @@ def yoloV9_func(image: gr.inputs.Image = None,
 
 
 inputs = [
-    gr.inputs.Image(type="filepath", label="Input Image"),
-    gr.inputs.Slider(minimum=320, maximum=1280, default=640,
+    gr.Image(type="filepath", label="Input Image"),
+    gr.Slider(minimum=320, maximum=1280, default=640,
                      step=32, label="Image Size"),
-    gr.inputs.Slider(minimum=0.0, maximum=1.0, default=0.25,
+    gr.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,
+    gr.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")
+outputs = gr.Image(type="filepath", label="Output Image")
 
 title = "CUSTOM yolov9 model for room cleanliness"