vumichien commited on
Commit
ced3dcd
·
1 Parent(s): 887cb19

remove ファイル名

Browse files
Files changed (1) hide show
  1. 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["標準科目"]