Johan713 commited on
Commit
64b9aa5
1 Parent(s): 117e9a7

Update nexus.py

Browse files
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
  """