ranavikas Johan713 commited on
Commit
25d74ba
1 Parent(s): 117e9a7

Update nexus.py (#1)

Browse files

- Update nexus.py (64b9aa5f8a98169b18e8b0d02383078b64cd0d47)


Co-authored-by: Jayash Bhardwaj <Johan713@users.noreply.huggingface.co>

Files changed (1) hide show
  1. nexus.py +2 -2
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
  """