Spaces:
Sleeping
Sleeping
Update classification.py
Browse files- classification.py +1 -1
classification.py
CHANGED
@@ -87,7 +87,7 @@ def save_data(df, filename):
|
|
87 |
df['Topic'] = df['Topic'].apply(lambda x: ', '.join(x) if isinstance(x, list) else x)
|
88 |
df['Score'] = df['Score'].apply(lambda x: ', '.join(map(str, x)) if isinstance(x, list) else x)
|
89 |
|
90 |
-
|
91 |
new_filename = filename.replace(".", "_classified.")
|
92 |
df.to_excel(new_filename, index=False)
|
93 |
return new_filename
|
|
|
87 |
df['Topic'] = df['Topic'].apply(lambda x: ', '.join(x) if isinstance(x, list) else x)
|
88 |
df['Score'] = df['Score'].apply(lambda x: ', '.join(map(str, x)) if isinstance(x, list) else x)
|
89 |
|
90 |
+
df = df.drop(columns=['Embeddings'])
|
91 |
new_filename = filename.replace(".", "_classified.")
|
92 |
df.to_excel(new_filename, index=False)
|
93 |
return new_filename
|