VyLala commited on
Commit
08ddb3f
·
verified ·
1 Parent(s): 39ad3dc

Update pipeline.py

Browse files
Files changed (1) hide show
  1. pipeline.py +2 -2
pipeline.py CHANGED
@@ -588,13 +588,13 @@ async def pipeline_with_gemini(accessions,stop_flag=None, save_df=None, niche_ca
588
  data_preprocess.save_text_to_docx(all_output, file_all_path)
589
  result_all_upload = upload_file_to_drive(file_all_path, all_filename, sample_folder_id)
590
  print("UPLOAD RESULT FOR all_output: ", result_all_upload)
591
- print(f"🔗 Uploaded file: https://drive.google.com/file/d/{result_chunk_upload}/view")
592
 
593
  acc_prompts = {acc: text}
594
  print("start model")
595
  predicted_output_info = await model.query_document_info(
596
  niche_cases=niche_cases,
597
- saveLinkFolder=saveLinkFolder,
598
  llm_api_function=model.call_llm_api,
599
  prompts=acc_prompts)
600
  for output_acc in predicted_output_info:
 
588
  data_preprocess.save_text_to_docx(all_output, file_all_path)
589
  result_all_upload = upload_file_to_drive(file_all_path, all_filename, sample_folder_id)
590
  print("UPLOAD RESULT FOR all_output: ", result_all_upload)
591
+ print(f"🔗 Uploaded file: https://drive.google.com/file/d/{result_all_upload}/view")
592
 
593
  acc_prompts = {acc: text}
594
  print("start model")
595
  predicted_output_info = await model.query_document_info(
596
  niche_cases=niche_cases,
597
+ saveLinkFolder=sample_folder_id,
598
  llm_api_function=model.call_llm_api,
599
  prompts=acc_prompts)
600
  for output_acc in predicted_output_info: