Update app.py

app.py

@@ -36,12 +36,15 @@ def respond(
     selected_dfs = torch.load('selected_dfs.sav', map_location=torch.device('cpu'))
     for df in selected_dfs:
         df = df.dropna()
-        df = df.sample(math.floor(df.shape[0]/10))
+        n = math.floor(df.shape[0]/1000)
+        print(n)
+        df = df.sample(n)
         for index, row in df.iterrows():
             print(index)
             print(row['Column1.question'])
-            #messages.append({"role": "user", "content": row['Column1.question']})
-            #messages.append({"role": "assistant", "content": row['Column1.answer']})
+            print(row['Column1.answer'])
+            messages.append({"role": "user", "content": row['Column1.question']})
+            messages.append({"role": "assistant", "content": row['Column1.answer']})
     
     messages.append({"role": "user", "content": message})
     print(messages)