Update app.py

app.py

@@ -123,7 +123,7 @@ def evaluate(
     max_new_tokens=128,
     **kwargs,
 ):
-    prompt = _TEMPLATE + "\n" + table + "\n" + "Q: " + question
+    prompt = _TEMPLATE + "\n" + _add_markup(table) + "\n" + "Q: " + question + "\n" + "A:"
     inputs = tokenizer(prompt, return_tensors="pt")
     input_ids = inputs["input_ids"].to(device)
     generation_config = GenerationConfig(
@@ -143,7 +143,7 @@ def evaluate(
         )
     s = generation_output.sequences[0]
     output = tokenizer.decode(s)
-    #return output.split("### Response:")[1].strip()
+    #return output.split("A:")[-1].strip()
     return output