Update app.py

app.py

@@ -18,8 +18,10 @@ from yolov5.models.yolo import Model
 from yolov5.utils.general import non_max_suppression
 
 device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
+print(f"Using device: {device}")
 model = torch.hub.load('ultralytics/yolov5', 'yolov5s', pretrained=True).to(device)
 model.eval()
+print("Model loaded successfully")
 
 def preprocess_image(image):
     try: