wangjin2000 commited on
Commit
86e36a2
·
verified ·
1 Parent(s): fb8afd6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -243,7 +243,7 @@ def predict_peptide_from_file(base_model_path, finetuned_model_path, file_obj, p
243
 
244
  results_df.to_csv(outpath,header=True, index=False)
245
 
246
- return output, outpath
247
 
248
  def suggest(option):
249
  if option == "Protein:P63279":
@@ -367,11 +367,11 @@ with demo:
367
  outputs = [output_text, input_seq],
368
  )
369
 
370
- # "Predict peptide sequence" actions
371
  predict_file_btn.click(
372
  fn = predict_peptide_from_file,
373
  inputs=[base_model_name,PEFT_model_name,uploaded_file,peptide_length,num_pred_peptides],
374
- outputs = [output_text, input_seq],
375
  )
376
 
377
  # "Finetune Pre-trained Model" actions
 
243
 
244
  results_df.to_csv(outpath,header=True, index=False)
245
 
246
+ return outpath
247
 
248
  def suggest(option):
249
  if option == "Protein:P63279":
 
367
  outputs = [output_text, input_seq],
368
  )
369
 
370
+ # "Predict peptide from a local file" actions
371
  predict_file_btn.click(
372
  fn = predict_peptide_from_file,
373
  inputs=[base_model_name,PEFT_model_name,uploaded_file,peptide_length,num_pred_peptides],
374
+ outputs = [output_file],
375
  )
376
 
377
  # "Finetune Pre-trained Model" actions