Spaces:
Paused
Paused
Update app.py
Browse files
app.py
CHANGED
|
@@ -250,8 +250,9 @@ with demo:
|
|
| 250 |
label="Finetuned Model Name",
|
| 251 |
interactive = True,
|
| 252 |
)
|
| 253 |
-
|
| 254 |
-
|
|
|
|
| 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 |
|