file added

app.py

@@ -46,7 +46,7 @@ sentiment_analyzer = pipeline("sentiment-analysis")
 
 # Load pre-trained YOLOv5 model
 def load_yolov5_model():
-    model = torch.hub.load(r"ultralytics/yolov5", 'custom', path=r'models\best.pt')
+    model = torch.hub.load(r"ultralytics/yolov5", 'custom', path=r'./models/best.pt')
     model.eval()
     return model
 

pred.py

@@ -8,7 +8,7 @@ pathlib.PosixPath = pathlib.WindowsPath
 
 
 
-model = torch.hub.load(r"ultralytics/yolov5", 'custom', path=r'models\best.pt')
+model = torch.hub.load(r"ultralytics/yolov5", 'custom', path=r'./models/best.pt')
 
 img_path = r"ACNE.jpg"
 image = Image.open(img_path)