Update app.py

app.py

@@ -6,12 +6,15 @@ import plotly.graph_objects as go
 import torch
 import gradio as gr
 import os
-from scipy.interpolate import interp1d
 from scipy.optimize import curve_fit
+import sys
 
-# Load YOLOv5 model from yolov5 repository
-from yolov5.models.experimental import attempt_load
-from yolov5.utils.general import non_max_suppression, xywh2xyxy
+# Add yolov5 directory to sys.path
+sys.path.append(os.path.join(os.path.dirname(__file__), "yolov5"))
+
+# Import YOLOv5 modules
+from models.experimental import attempt_load
+from utils.general import non_max_suppression, xywh2xyxy
 
 # Cricket pitch dimensions (in meters)
 PITCH_LENGTH = 20.12  # Length of cricket pitch (stumps to stumps)