Update app.py

app.py

@@ -3,12 +3,12 @@ import torch
 import cv2
 import numpy as np
 import matplotlib.pyplot as plt
+from yolov5 import YOLOv5
 
-# Load the trained YOLOv5 model (best.pt) from a local path
-model = torch.load("best.pt", map_location="cpu")  # Adjust the path as needed
-model.eval()  # Put the model in evaluation mode
+# Load the trained YOLOv5 model (best.pt) using the YOLOv5 library
+model = YOLOv5("best.pt")  # This will use YOLOv5's internal loading mechanism
 
-# Function to process video and calculate ball trajectory, speed, and visualize the pitch
+# Function to process the video and calculate ball trajectory, speed, and visualize the pitch
 def process_video(video_file):
     # Load video file using OpenCV
     video = cv2.VideoCapture(video_file.name)