H2H-eval-comparator

Sleeping

rohansampath commited on Feb 11

Commit

7aad8f7

verified ·

1 Parent(s): 792f192

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -65,7 +65,7 @@ def generate_answer(question):
     model, tokenizer = load_model()
     # Mistral instruction format
-    prompt = f"""<s>[INST] {question}. Provide only the answer. [/INST]"""
     inputs = tokenizer(prompt, return_tensors="pt").to('cuda')
     with torch.no_grad():

     model, tokenizer = load_model()
     # Mistral instruction format
+    prompt = f"""<s>[INST] {question}. Provide only the numerical answer. [/INST]"""
     inputs = tokenizer(prompt, return_tensors="pt").to('cuda')
     with torch.no_grad():