Update app.py

app.py

@@ -7,7 +7,7 @@ import random
 
 # Load YOLOv5 model
 device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
-model = torch.hub.load('ultralytics/yolov5', 'yolov5x', pretrained=True).to(device) 
+model = torch.hub.load('ultralytics/yolov5', 'yolov5s', pretrained=True).to(device) 
 
 # Get class names from the model
 CLASS_NAMES = model.names