heymenn commited on
Commit
267ff77
1 Parent(s): 4e2a30f

Update excel_chat.py

Browse files
Files changed (1) hide show
  1. excel_chat.py +2 -2
excel_chat.py CHANGED
@@ -237,7 +237,7 @@ def chat_with_mistral(source_cols, dest_col, prompt, excel_file, url, search_col
237
  try:
238
  if cpt == checkpoint:
239
  df2 = df
240
- df2.to_excel("checkpointfile", index=False)
241
 
242
  checkpoint += 1
243
  except Exception as e:
@@ -257,7 +257,7 @@ def chat_with_mistral(source_cols, dest_col, prompt, excel_file, url, search_col
257
  with zipfile.ZipFile(zip_file_path, 'w') as zipf:
258
  zipf.write(conf_file_path, os.path.basename(conf_file_path))
259
 
260
- return file_name, df.head(5), new_prompts, new_keywords, new_user, zip_file_path, "checkpointfile"
261
 
262
 
263
  def get_columns(file,progress=gr.Progress()):
 
237
  try:
238
  if cpt == checkpoint:
239
  df2 = df
240
+ df2.to_excel("checkpointfile.xlsx", index=False)
241
 
242
  checkpoint += 1
243
  except Exception as e:
 
257
  with zipfile.ZipFile(zip_file_path, 'w') as zipf:
258
  zipf.write(conf_file_path, os.path.basename(conf_file_path))
259
 
260
+ return file_name, df.head(5), new_prompts, new_keywords, new_user, zip_file_path, "checkpointfile.xlsx"
261
 
262
 
263
  def get_columns(file,progress=gr.Progress()):