Update app.py

app.py

@@ -4,8 +4,8 @@ import gradio as gr
 from PIL import Image
 
 # Step 1: Load the custom YOLOv5 model
-# Adjust the model path based on your deployment setup (Hugging Face model hub or local path)
-model = torch.hub.load('ultralytics/yolov5', 'custom', path='best.pt')  # If you use Hugging Face model hub, use the correct identifier
+# Make sure the path to the model is correct, either locally or from Hugging Face model hub
+model = torch.hub.load('ultralytics/yolov5', 'custom', path='best.pt')  # Adjust the path if needed
 
 # Step 2: Define weapon classes to detect
 weapon_classes = ['pistol', 'gun', 'rifle', 'shotgun', 'handgun']