wangjin2000 commited on
Commit
d319533
·
verified ·
1 Parent(s): 6ffebf7

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -0
app.py CHANGED
@@ -250,12 +250,17 @@ with demo:
250
  label="Finetuned Model Name",
251
  interactive = True,
252
  )
 
 
253
  with gr.Column(scale=5, variant="compact"):
254
  name = gr.Dropdown(
255
  label="Choose a Sample Protein",
256
  value="Protein:P63279",
257
  choices=["Default protein", "Antifreeze protein", "Protein:P63279", "AI Generated protein", "7-bladed propeller fold", "custom"]
258
  )
 
 
 
259
  gr.Markdown(
260
  "## Predict peptide sequence and Plot structure for targeted protein sequence and the binder:"
261
  )
 
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",
258
  value="Protein:P63279",
259
  choices=["Default protein", "Antifreeze protein", "Protein:P63279", "AI Generated protein", "7-bladed propeller fold", "custom"]
260
  )
261
+ with gr.Row():
262
+ guidance_scale=gr.Slider(minimum=0, maximum=20, step=0.1, label="Guidance Scale", value=7.5)
263
+ num_sampling_steps=gr.Slider(minimum=10, maximum=100, step=1, label="Number of Sampling Steps", value=20)
264
  gr.Markdown(
265
  "## Predict peptide sequence and Plot structure for targeted protein sequence and the binder:"
266
  )