Bofandra commited on
Commit
6e9eef1
·
verified ·
1 Parent(s): 8da225e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -36,7 +36,7 @@ def respond(
36
  selected_dfs = torch.load('selected_dfs.sav', map_location=torch.device('cpu'))
37
  for df in selected_dfs:
38
  df = df.dropna()
39
- n = math.floor(df.shape[0]/1000)
40
  print(n)
41
  df = df.sample(n)
42
  for index, row in df.iterrows():
@@ -47,7 +47,7 @@ def respond(
47
  messages.append({"role": "assistant", "content": row['Column1.answer']})
48
 
49
  messages.append({"role": "user", "content": message})
50
- print(messages)
51
 
52
  response = ""
53
 
 
36
  selected_dfs = torch.load('selected_dfs.sav', map_location=torch.device('cpu'))
37
  for df in selected_dfs:
38
  df = df.dropna()
39
+ n = math.floor(df.shape[0]/10000)
40
  print(n)
41
  df = df.sample(n)
42
  for index, row in df.iterrows():
 
47
  messages.append({"role": "assistant", "content": row['Column1.answer']})
48
 
49
  messages.append({"role": "user", "content": message})
50
+ #print(messages)
51
 
52
  response = ""
53