Bofandra commited on
Commit
bdaa841
1 Parent(s): 20d2a18

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -35,6 +35,7 @@ def respond(
35
  selected_dfs = torch.load('selected_dfs.sav', map_location=torch.device('cpu'))
36
  for df in selected_dfs:
37
  df = df.dropna()
 
38
  for index, row in df.iterrows():
39
  messages.append({"role": "user", "content": row['Column1.question']})
40
  messages.append({"role": "assistant", "content": row['Column1.answer']})
 
35
  selected_dfs = torch.load('selected_dfs.sav', map_location=torch.device('cpu'))
36
  for df in selected_dfs:
37
  df = df.dropna()
38
+ df = df.sample(df.shape[0].div(10))
39
  for index, row in df.iterrows():
40
  messages.append({"role": "user", "content": row['Column1.question']})
41
  messages.append({"role": "assistant", "content": row['Column1.answer']})