Update app.py

app.py

@@ -31,7 +31,7 @@ BATCH_SIZE = 4  # Process 4 frames at a time
 
 # Load model
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
-model = attempt_load("best.pt")  # Load without map_location
+model = attempt_load("yolov5s.pt")  # Load without map_location
 model.to(device).eval()  # Move model to device and set to evaluation mode
 
 # Function to process video and detect ball