Spaces:

Reshmarb
/

chatbot2

Sleeping

Reshmarb commited on Jan 27

Commit

cae54fd

1 Parent(s): 989bc70

file added

Files changed (1) hide show

app.py CHANGED Viewed

@@ -47,9 +47,9 @@ sentiment_analyzer = pipeline("sentiment-analysis")
 # Load pre-trained YOLOv5 model
 def load_yolov5_model():
     model = torch.hub.load(
-        r'C:\Users\RESHMA R B\OneDrive\Documents\Desktop\project_without_malayalam\chatbot2\yolov5',
         'custom',
-        path=r"C:\Users\RESHMA R B\OneDrive\Documents\Desktop\project_without_malayalam\chatbot2\models\best.pt",
         source="local"
     )
     model.eval()

 # Load pre-trained YOLOv5 model
 def load_yolov5_model():
     model = torch.hub.load(
+        r'yolov5',
         'custom',
+        path=r"models\best.pt",
         source="local"
     )
     model.eval()