Update app.py

app.py

@@ -61,7 +61,7 @@ def get_llama_response(query):
         max_new_tokens=128,
         return_full_text=False,
     )
-    outputs = pipe(prompt)
+    outputs = pipe(query)
     return outputs[0]["generated_text"]
 
 # --- PDF Generation ---