Update app.py

app.py

@@ -34,8 +34,8 @@ 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['user']})
-        messages.append({"role": "assistant", "content": df['assistant']})
+        messages.append({"role": "user", "content": df['Column1.question']})
+        messages.append({"role": "assistant", "content": df['Column1.answer']})
     
     messages.append({"role": "user", "content": message})
     print(messages)