heymenn commited on
Commit
8931e44
1 Parent(s): 1260e1e

fix error for checkpoint retrieval (#5)

Browse files

- fix error for checkpoint retrieval (38ea626e8ab29cc6f5dd555667f0ec58c69ede30)

Files changed (1) hide show
  1. excel_chat.py +2 -2
excel_chat.py CHANGED
@@ -236,7 +236,7 @@ def chat_with_mistral(source_cols, dest_col, prompt, excel_file, url, search_col
236
 
237
  try:
238
  if cpt == checkpoint:
239
- save_name = user['save_name']+".xlsx"
240
  df.to_excel(save_name, index=False)
241
  checkpoint += 50
242
 
@@ -247,7 +247,7 @@ def chat_with_mistral(source_cols, dest_col, prompt, excel_file, url, search_col
247
  # progress((index+1)/len(df),desc=f'Request {index+1}/{len(df)}!')
248
 
249
  df.to_excel(file_name, index=False)
250
- save_name = user['save_name']+".xlsx"
251
  df.to_excel(save_name, index=False)
252
 
253
  zip_file_path = 'config_file.zip'
 
236
 
237
  try:
238
  if cpt == checkpoint:
239
+ save_name = user['save_name']
240
  df.to_excel(save_name, index=False)
241
  checkpoint += 50
242
 
 
247
  # progress((index+1)/len(df),desc=f'Request {index+1}/{len(df)}!')
248
 
249
  df.to_excel(file_name, index=False)
250
+ save_name = user['save_name']
251
  df.to_excel(save_name, index=False)
252
 
253
  zip_file_path = 'config_file.zip'