Update nexus.py (#3)

- Update nexus.py (ad8eb420d63a225c1e4502358f9f64f9d66ca14e)


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

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]]:
     """