Update app.py

app.py

@@ -1,20 +1,24 @@
 import gradio as gr
 from ultralytics import YOLO
+from PIL import Image
 
 # Load YOLO model
 model = YOLO("Suspicious_Activities_nano.pt")
 
-# Define a function to predict and return the result
+# Define the prediction function
 def predict_suspicious_activity(image):
-    results = model.predict(source=image, show=True, conf=0.6)
-    return results.pandas().xywh[0].to_dict()  # Adjust as per your model output structure
+    results = model.predict(source=image, show=False, conf=0.6)
+    results_img = results[0].plot()  # Get the image with bounding boxes/annotations
+    return Image.fromarray(results_img)  # Convert NumPy array to PIL Image for Gradio
 
 # Create Gradio interface
 interface = gr.Interface(
-    fn=predict_suspicious_activity,
-    inputs=gr.inputs.Image(type="pil"),  # Input type for uploading images
-    outputs="json"  # Output type (you can adjust this based on your needs)
+    fn=predict_suspicious_activity,  # Function to run on input
+    inputs=gr.Image(type="pil"),  # Input type is image (PIL)
+    outputs=gr.Image(type="pil"),  # Output type is image (PIL)
+    title="Suspicious Activity Detection with YOLO",  # Interface title
+    description="Upload an image to detect suspicious activities.",  # Description
 )
 
 # Launch the interface
-interface.launch()
+interface.launch(share=True)