Update app.py

app.py

@@ -24,16 +24,7 @@ def predict_image(input_image):
     img_bgr = result[0].plot()
     img_rgb = cv2.cvtColor(img_bgr, cv2.COLOR_BGR2RGB)  # Convert BGR to RGB
     output_image = Image.fromarray(img_rgb)  # Use the RGB image for output
-
-    # Get the original filename (without path)
-    original_filename = os.path.basename(input_image.name)
-    
-    # Get the name without the extension and append '_detected'
-    base_filename, _ = os.path.splitext(original_filename)
-    output_image_jpg = base_filename + "_detected.jpg"
-    output_image.save(output_image_jpg, "JPEG")
-    
-    return output_image_jpg # Return path to saved JPG image
+    return output_image
 
 # Video prediction function
 def predict_video(input_video):
@@ -83,8 +74,8 @@ with gr.Blocks() as app:
     with gr.Tabs():
         with gr.Tab("Image Detection"):
             gr.Interface(fn=predict_image,
-                         inputs=gr.File(type="pil", label="Upload Image"),
-                         outputs=gr.File(type="pil", label="Download Image"),
+                         inputs=gr.Image(type="pil", label="Upload Image"),
+                         outputs=gr.Image(type="pil", label="Download Image"),
                          title="Image Object Detection",
                          description="Upload an image to detect Wildebeest and/or Hyena.")