Update app.py

app.py

@@ -10,12 +10,15 @@ torch.hub.download_url_to_file(
     'https://s.wsj.net/public/resources/images/OB-EQ974_diwali_G_20091014112400.jpg', 'three.jpg')
 
 
-def handle_classify(image: gr.inputs.Image = None):
+def handle_classify(image=None):
     """This function performs YOLOv8 object detection on the given image.
 
     Args:
         image (gr.inputs.Image, optional): Input image to detect objects on. Defaults to None.
     """
+    
+    if not image:
+        return "No image found"
 
     model_path = "racist2.0.pt"
     model = YOLO(model_path)
@@ -33,11 +36,11 @@ def handle_classify(image: gr.inputs.Image = None):
 
 
 inputs = [
-    gr.inputs.Image(type="filepath", label="Input Image"),
+    gr.Image(type="filepath", label="Input Image"),
 ]
 
 
-outputs = gr.outputs.Text()
+outputs = gr.Textbox()
 
 title = "Racist model v2"