Spaces:
Running
Running
Update app.py
Browse files
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']})
|