Spaces:
Sleeping
Sleeping
Update nexus.py (#1)
Browse files- Update nexus.py (64b9aa5f8a98169b18e8b0d02383078b64cd0d47)
Co-authored-by: Jayash Bhardwaj <Johan713@users.noreply.huggingface.co>
nexus.py
CHANGED
@@ -28,8 +28,8 @@ endpoint = 'https://api.together.xyz/inference'
|
|
28 |
|
29 |
# Load the hallucination evaluation model
|
30 |
model_name = "vectara/hallucination_evaluation_model"
|
31 |
-
model = AutoModelForSequenceClassification.from_pretrained(model_name)
|
32 |
-
tokenizer = AutoTokenizer.from_pretrained(model_name)
|
33 |
|
34 |
def search_pubmed(query: str) -> Optional[List[str]]:
|
35 |
"""
|
|
|
28 |
|
29 |
# Load the hallucination evaluation model
|
30 |
model_name = "vectara/hallucination_evaluation_model"
|
31 |
+
model = AutoModelForSequenceClassification.from_pretrained(model_name, trust_remote_code=True)
|
32 |
+
tokenizer = AutoTokenizer.from_pretrained(model_name, trust_remote_code=True)
|
33 |
|
34 |
def search_pubmed(query: str) -> Optional[List[str]]:
|
35 |
"""
|