Update app.py

app.py

@@ -34,8 +34,9 @@ def respond(
 
     selected_dfs = torch.load('selected_dfs.sav', map_location=torch.device('cpu'))
     for df in selected_dfs:
-        messages.append({"role": "user", "content": df['Column1.question']})
-        messages.append({"role": "assistant", "content": df['Column1.answer']})
+        for index, row in df.iterrows():
+            messages.append({"role": "user", "content": row['Column1.question']})
+            messages.append({"role": "assistant", "content": row['Column1.answer']})
     
     messages.append({"role": "user", "content": message})
     print(messages)