Update app.py

app.py

@@ -153,7 +153,7 @@ class DocumentRetrievalAndGeneration:
         </s>
         """
         messages = [{"role": "system", "content": prompt}]
-        messages.append({"role": "user", "content": message})
+        messages.append({"role": "user", "content": query})
         response = ""
 
         for message in client.chat_completion(messages,max_tokens=2048,stream=True,temperature=0.7):