Update nexus.py

nexus.py

@@ -29,7 +29,13 @@ endpoint = 'https://api.together.xyz/inference'
 # Load the hallucination evaluation model
 model_name = "vectara/hallucination_evaluation_model"
 model = AutoModelForSequenceClassification.from_pretrained(model_name, trust_remote_code=True)
-tokenizer = AutoTokenizer.from_pretrained(model_name, trust_remote_code=True)
+tokenizer = AutoTokenizer.from_pretrained("bert-base-uncased")
+
+def vectara_hallucination_evaluation_model(text):
+    inputs = tokenizer(text, return_tensors="pt", truncation=True, max_length=512)
+    outputs = model(**inputs)
+    hallucination_probability = outputs.logits[0][0].item()  
+    return hallucination_probability
 
 def search_pubmed(query: str) -> Optional[List[str]]:
     """