Made changes to the prompt

app.py

@@ -28,6 +28,13 @@ model.eval()
 
 
 def answer_question(prompt, temperature, top_k, max_length):
+
+    prompt = f'''
+    ### Instruction: You are a helpful AI Assistant for question answering.
+    ### Input : {prompt}
+
+    ### Response: 
+    '''
     with torch.no_grad():
         generated_text = topk_sampling(model, prompt, max_length=max_length, top_k=top_k, temperature=temperature, device=ModelArgs.device)
         # print("Gnerated: ", generated_text)