Update app.py

app.py

@@ -23,7 +23,7 @@ PITCH_WIDTH = 3.05    # Width of pitch
 STUMP_HEIGHT = 0.71   # Stump height
 STUMP_WIDTH = 0.2286  # Stump width (including bails)
 
-# Model input size (adjust if best.pt was trained with a different size)
+# Model input size (adjust if yolov5s.pt was trained with a different size)
 MODEL_INPUT_SIZE = (640, 640)  # (height, width)
 FRAME_SKIP = 2  # Process every 2nd frame
 MIN_DETECTIONS = 10  # Stop after 10 detections
@@ -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("yolov5s.pt")  # Load without map_location
+model = attempt_load("yolov5s.pt")  # Load yolov5s.pt instead of best.pt
 model.to(device).eval()  # Move model to device and set to evaluation mode
 
 # Function to process video and detect ball