jarguello76 commited on
Commit
aee2e55
·
verified ·
1 Parent(s): 48dd983

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -26,7 +26,7 @@ class QuestionValidation:
26
  self.client = InferenceClient(model="HuggingFaceH4/zephyr-7b-beta", token=HF_TOKEN)
27
 
28
  self.llm2_model = "HuggingFaceH4/zephyr-7b-beta"
29
- self.embedding_model = SentenceTransformer("sentence-transformers/all-mpnet-base-v2")
30
 
31
  def guess_question(self, answer: str) -> str:
32
  prompt = f"This was the answer: {answer}\nWhat question would likely have led to it?"
 
26
  self.client = InferenceClient(model="HuggingFaceH4/zephyr-7b-beta", token=HF_TOKEN)
27
 
28
  self.llm2_model = "HuggingFaceH4/zephyr-7b-beta"
29
+ self.embedding_model = SentenceTransformer("sentence-transformers/all-MiniLM-L6-v2")
30
 
31
  def guess_question(self, answer: str) -> str:
32
  prompt = f"This was the answer: {answer}\nWhat question would likely have led to it?"