Update app.py

app.py

@@ -6,19 +6,21 @@ from ultralytics import YOLO
 from huggingface_hub import hf_hub_download
 import os
 
-# Verify that Hugging Face repo and file paths are correct
-REPO_ID = "StephanST/WALDO30"  # Update if the repository ID is different
-MODEL_FILENAME = "WALDO30_yolov8m_640x640.pt"
+# Verify paths and Hugging Face repository details
+REPO_ID = "StephanST/WALDO30"  # Replace with the correct repo ID if different
+MODEL_FILENAME = "WALDO30_yolov8m_640x640.pt"  # Replace if the filename is different
 
 # Download the model from Hugging Face
 try:
     model_path = hf_hub_download(repo_id=REPO_ID, filename=MODEL_FILENAME)
+    print(f"Model downloaded successfully to: {model_path}")
 except Exception as e:
     raise RuntimeError(f"Failed to download model from Hugging Face. Verify `repo_id` and `filename`. Error: {e}")
 
 # Load the YOLOv8 model
 try:
-    model = YOLO(model_path)  # Ensure the model path is correct
+    model = YOLO(model_path)  # Load the YOLOv8 model
+    print("Model loaded successfully!")
 except Exception as e:
     raise RuntimeError(f"Failed to load the YOLO model. Verify the model file at `{model_path}`. Error: {e}")
 
@@ -54,19 +56,30 @@ def detect_on_video(video):
     except Exception as e:
         raise RuntimeError(f"Error during video processing: {e}")
 
-# Gradio Interface
-image_input = gr.Image(type="pil", label="Upload Image")
-video_input = gr.Video(label="Upload Video")  # Removed invalid `type` argument
-image_output = gr.Image(type="pil", label="Detected Image")
-video_output = gr.Video(label="Detected Video")
+# Gradio Interface using Blocks
+with gr.Blocks() as app:
+    gr.Markdown("# WALDO30 YOLOv8 Object Detection")
+    gr.Markdown("Upload an image or video to see object detection results using the WALDO30 YOLOv8 model.")
 
-app = gr.Interface(
-    fn=[detect_on_image, detect_on_video],
-    inputs=[image_input, video_input],
-    outputs=[image_output, video_output],
-    title="WALDO30 YOLOv8 Object Detection",
-    description="Upload an image or video to see object detection results using the WALDO30 YOLOv8 model."
-)
+    # Image processing block
+    with gr.Row():
+        with gr.Column():
+            image_input = gr.Image(type="pil", label="Upload Image")
+            image_button = gr.Button("Detect on Image")
+        with gr.Column():
+            image_output = gr.Image(type="pil", label="Detected Image")
+
+    # Video processing block
+    with gr.Row():
+        with gr.Column():
+            video_input = gr.Video(label="Upload Video")
+            video_button = gr.Button("Detect on Video")
+        with gr.Column():
+            video_output = gr.Video(label="Detected Video")
+
+    # Set up events
+    image_button.click(detect_on_image, inputs=image_input, outputs=image_output)
+    video_button.click(detect_on_video, inputs=video_input, outputs=video_output)
 
 if __name__ == "__main__":
     app.launch()