Spaces:
Running
Running
add new column to keep
Browse files- routes/predict.py +1 -1
routes/predict.py
CHANGED
@@ -62,7 +62,7 @@ async def predict(
|
|
62 |
df_predicted = nameMappingHelper.map_standard_names()
|
63 |
|
64 |
# Create output dataframe and save to CSV
|
65 |
-
column_to_keep = ['シート名', '行', '科目', '分類', '名称', '摘要', '備考']
|
66 |
output_df = inputData.dataframe[column_to_keep].copy()
|
67 |
output_df.reset_index(drop=False, inplace=True)
|
68 |
output_df.loc[:, "出力_科目"] = df_predicted["出力_科目"]
|
|
|
62 |
df_predicted = nameMappingHelper.map_standard_names()
|
63 |
|
64 |
# Create output dataframe and save to CSV
|
65 |
+
column_to_keep = ['シート名', '行', '科目', '中科目', '分類', '名称', '摘要', '備考']
|
66 |
output_df = inputData.dataframe[column_to_keep].copy()
|
67 |
output_df.reset_index(drop=False, inplace=True)
|
68 |
output_df.loc[:, "出力_科目"] = df_predicted["出力_科目"]
|