Update app.py

app.py

@@ -3,11 +3,9 @@ import torch
 import cv2
 from PIL import Image
 import numpy as np
-import tempfile
-
 
 # YOLOv5 Model Loading (best.pt)
-@st.cache_resource  # Cache the model for efficiency
+@st.cache(allow_output_mutation=True)  # Cache the model for efficiency
 def load_model():
     return torch.hub.load('ultralytics/yolov5', 'custom', path='best.pt', force_reload=True) 
 
@@ -43,7 +41,7 @@ if upload_option == "Real-Time Webcam":
             ret, frame = cap.read()
             if not ret:
                 st.warning("Error: Unable to capture frame. Please check your webcam settings.")  # More informative message
-                continue  # Skip to the next iteration instead of stopping the loop entirely
+                break  # Exit the loop if frame capture fails
 
             # Convert to RGB and detect objects
             frame_rgb = cv2.cvtColor(frame, cv2.COLOR_BGR2RGB)
@@ -51,6 +49,6 @@ if upload_option == "Real-Time Webcam":
 
             # Render and display results
             annotated_frame = results.render()[0]
-            FRAME_WINDOW.image(annotated_frame, channels="BGR")  # Display in BGR format
+            FRAME_WINDOW.image(annotated_frame, channels="RGB")  # Display in RGB format
 
         cap.release()