Update app.py

app.py

@@ -26,7 +26,21 @@ path = [['image_0.jpg'], ['image_1.jpg']]
 video_path = [['video.mp4']]
 
 def show_preds_image(image_path):
+    # Save the user-uploaded image to the local directory
+    save_path = 'uploaded_image.jpg'
+    if isinstance(image_path, str) and image_path.startswith('data:'):
+        # Convert base64 to image
+        import base64
+        from io import BytesIO
+        from PIL import Image
+        base64_str = image_path.split(',')[1]
+        image = Image.open(BytesIO(base64.b64decode(base64_str)))
+        image.save(save_path)
+        image_path = save_path
+
     image = cv2.imread(image_path)
+    if image is None:
+        raise ValueError(f"Image at path {image_path} could not be loaded.")
     outputs = model.predict(source=image_path)
     results = outputs[0].cpu().numpy()
     for i, det in enumerate(results.boxes.xyxy):
@@ -57,6 +71,8 @@ interface_image = gr.Interface(
 
 def show_preds_video(video_path):
     cap = cv2.VideoCapture(video_path)
+    if not cap.isOpened():
+        raise ValueError(f"Video at path {video_path} could not be loaded.")
     while cap.isOpened():
         ret, frame = cap.read()
         if ret: