Spaces:
Runtime error
Runtime error
removed token
Browse files- semantic_search.py +2 -2
semantic_search.py
CHANGED
@@ -82,8 +82,8 @@ class SemanticSearch:
|
|
82 |
|
83 |
def load_model(self):
|
84 |
if hf_token and hf_model_name:
|
85 |
-
self.tokenizer = AutoTokenizer.from_pretrained(hf_model_name, use_auth_token=True
|
86 |
-
self.model = AutoModel.from_pretrained(hf_model_name, use_auth_token=True
|
87 |
else:
|
88 |
self.tokenizer = AutoTokenizer.from_pretrained(global_model_path)
|
89 |
self.model = AutoModel.from_pretrained(global_model_path).to(self.device)
|
|
|
82 |
|
83 |
def load_model(self):
|
84 |
if hf_token and hf_model_name:
|
85 |
+
self.tokenizer = AutoTokenizer.from_pretrained(hf_model_name, use_auth_token=True)
|
86 |
+
self.model = AutoModel.from_pretrained(hf_model_name, use_auth_token=True).to(self.device)
|
87 |
else:
|
88 |
self.tokenizer = AutoTokenizer.from_pretrained(global_model_path)
|
89 |
self.model = AutoModel.from_pretrained(global_model_path).to(self.device)
|