wangjin2000 commited on
Commit
9f3ddc0
·
verified ·
1 Parent(s): 03ec694

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -3
app.py CHANGED
@@ -250,8 +250,9 @@ with demo:
250
  label="Finetuned Model Name",
251
  interactive = True,
252
  )
253
- peptide_length=gr.Slider(minimum=10, maximum=100, step=1, label="Peptide Maximum Length", value=50)
254
- num_pred_peptides=gr.Slider(minimum=1, maximum=10, step=1, label="Number of Predicted Peptides Per Protein", value=4)
 
255
  with gr.Column(scale=5, variant="compact"):
256
  name = gr.Dropdown(
257
  label="Choose a Sample Protein",
@@ -315,7 +316,7 @@ with demo:
315
  # "Predict peptide sequence" actions
316
  predict_btn.click(
317
  fn = predict_peptide,
318
- inputs=[base_model_name,PEFT_model_name,input_seq],
319
  outputs = [output_text, input_seq],
320
  )
321
 
 
250
  label="Finetuned Model Name",
251
  interactive = True,
252
  )
253
+ with gr.Row():
254
+ peptide_length=gr.Slider(minimum=10, maximum=100, step=1, label="Peptide Maximum Length", value=50)
255
+ num_pred_peptides=gr.Slider(minimum=1, maximum=10, step=1, label="Number of Predicted Peptides Per Protein", value=4)
256
  with gr.Column(scale=5, variant="compact"):
257
  name = gr.Dropdown(
258
  label="Choose a Sample Protein",
 
316
  # "Predict peptide sequence" actions
317
  predict_btn.click(
318
  fn = predict_peptide,
319
+ inputs=[base_model_name,PEFT_model_name,input_seq,peptide_length],
320
  outputs = [output_text, input_seq],
321
  )
322