Update app.py

app.py

@@ -17,7 +17,7 @@ class RobustSafetyMonitor:
         self.colors = [(0, 0, 255), (255, 0, 0), (0, 255, 0), (255, 255, 0), (255, 0, 255)]
         
         # Load YOLOv5 model for general object detection
-        self.yolo_model = torch.hub.load('ultralytics/yolov5', 'yolov5s')
+        self.yolo_model = torch.hub.load('ultralytics/yolov5', 'yolov5s', pretrained=True)
 
     def preprocess_image(self, frame):
         """Process image for analysis."""
@@ -186,7 +186,7 @@ def create_monitor_interface():
                 print(f"Processing error: {str(e)}")
                 return None, f"Error processing image: {str(e)}"
             
-        input_image.change(
+        input_image.upload(
             fn=analyze_image,
             inputs=input_image,
             outputs=[output_image, analysis_text]