Update app.py

app.py

@@ -24,8 +24,8 @@ STUMP_WIDTH = 0.2286  # Stump width (including bails)
 
 # Load model
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
-model = attempt_load("best.pt", map_location=device)
-model.eval()
+model = attempt_load("best.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
 def process_video(video_path):