imravi commited on
Commit
e9c0f46
·
verified ·
1 Parent(s): ac4dd14

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +1 -1
main.py CHANGED
@@ -20,7 +20,7 @@ class PlantChatbot:
20
  self.device = "cuda" if torch.cuda.is_available() else "cpu"
21
 
22
  self.embedding_model = SentenceTransformer(
23
- model_name_or_path="all-mpnet-base-v2",
24
  device=self.device
25
  )
26
 
 
20
  self.device = "cuda" if torch.cuda.is_available() else "cpu"
21
 
22
  self.embedding_model = SentenceTransformer(
23
+ model_name_or_path="sentence-transformers/all-mpnet-base-v2",
24
  device=self.device
25
  )
26