Update app.py

app.py

@@ -163,9 +163,8 @@ def main(query: str, client: QdrantClient, collection_name: str, llm, dense_mode
               "content": f"Question: {query}"
           }
       ], stop=["</s>"], temperature=0.7)
-    print(response)
 
-    text = response["choices"][0]["text"]
+    text = response["choices"][0]["message"]['content']
     print(f'TEXT: {text}')
 
     prompt = f"""Q: Write a summary of the following text delimited by triple backquotes that includes the main points and any important details.
@@ -185,7 +184,7 @@ def main(query: str, client: QdrantClient, collection_name: str, llm, dense_mode
               "content": f"""Write a summary of the following text delimited by triple backquotes. Ensure the summary covers the key points of the text.
               ```{text}```"""
           }
-      ], stop=["</s>"], temperature=0.7, max_tokens=3000)['choices'][0]['text']
+      ], stop=["</s>"], temperature=0.7, max_tokens=3000)['choices'][0]['message']['content']
     print(f'OUTPUT: {output}')
     return output