Spaces:
Running
Running
remove ファイル名
Browse files- routes/predict.py +2 -1
routes/predict.py
CHANGED
@@ -59,7 +59,8 @@ async def predict(
|
|
59 |
raise HTTPException(status_code=500, detail=str(e))
|
60 |
|
61 |
# Create output dataframe and save to CSV
|
62 |
-
column_to_keep = ['ファイル名', 'シート名', '行', '科目', '中科目', '分類', '名称', '摘要', '備考']
|
|
|
63 |
output_df = inputData.dataframe[column_to_keep].copy()
|
64 |
output_df.reset_index(drop=False, inplace=True)
|
65 |
output_df.loc[:, "出力_科目"] = df_predicted["標準科目"]
|
|
|
59 |
raise HTTPException(status_code=500, detail=str(e))
|
60 |
|
61 |
# Create output dataframe and save to CSV
|
62 |
+
# column_to_keep = ['ファイル名', 'シート名', '行', '科目', '中科目', '分類', '名称', '摘要', '備考']
|
63 |
+
column_to_keep = ['シート名', '行', '科目', '中科目', '分類', '名称', '摘要', '備考']
|
64 |
output_df = inputData.dataframe[column_to_keep].copy()
|
65 |
output_df.reset_index(drop=False, inplace=True)
|
66 |
output_df.loc[:, "出力_科目"] = df_predicted["標準科目"]
|