Update app.py

app.py

@@ -54,6 +54,7 @@ def predict_structure(prefix, feature_dict, model_runners, random_seed=0):
     """Predicts structure using AlphaFold for the given sequence."""
 
     # Run the models.
+    # currently we only run model1
     plddts = {}
     for model_name, model_runner in model_runners.items():
         processed_feature_dict = model_runner.process_features(
@@ -328,15 +329,14 @@ with proteindream:
         with gr.Row():
             inp = gr.Textbox(placeholder="M", label="Start sequence")
             length = gr.Number(value=50, label="Target sequence length")
-        btn = gr.Button("Autocomplete sequences")
-
-    seqs = [
-        "MTAEADPAPLAANPPAPVRPIQFHDVSVRYEARPWLRALWDVASGSFIGLLGASGAGKSTCVDLLNGVRKPSSGERFVRGQPSRGRKGRFNRRVAMVFQDVRHQLFSRSVAREIAFGLENLPTSAAAIDRRVS",
-        "MTAGIVAGGIAGGVAGYKAKKHRKAVKATMIAAGVSGGIGGGYIGEKFNRRLAKHEDRVRRSAPRHKKHSSYSKSSGEGGGILGKLFGR",
-        "MTAVLVAIALEMQNPHRMALAAVLCGQFTVAVAAEPFAPEGVAEGLNPLGDLLAESPLLEVVSATLALLVALGTATSLSWISGPVSALPAPSFQSSETPYPQRPIERESFDQDSREEDPWDRL",
-        "MTARVRNRSSSRSYVLDFADLADGQREVLLPESRGNASEVDLPAGTTVNVTIDVTASGTGTLTARTPDGADVVSNEYELTVERDTDLTRVETESPQVAAGETATVTGTAENVGTVAGEREVTAYVDGE",
-        "MTAAGWREEGTPFARIARQLGRHVTSVRQAAGRVRQQMGLTSPDPADPPRSGPTPTIPIEQERA",
-    ]
+        btn = gr.Button("Predict sequences using protGPT2")
+
+    seqs = []
+    #        "MTAEADPAPLAANPPAPVRPIQFHDVSVRYEARPWLRALWDVASGSFIGLLGASGAGKSTCVDLLNGVRKPSSGERFVRGQPSRGRKGRFNRRVAMVFQDVRHQLFSRSVAREIAFGLENLPTSAAAIDRRVS",
+    #        "MTAGIVAGGIAGGVAGYKAKKHRKAVKATMIAAGVSGGIGGGYIGEKFNRRLAKHEDRVRRSAPRHKKHSSYSKSSGEGGGILGKLFGR",
+    #        "MTAVLVAIALEMQNPHRMALAAVLCGQFTVAVAAEPFAPEGVAEGLNPLGDLLAESPLLEVVSATLALLVALGTATSLSWISGPVSALPAPSFQSSETPYPQRPIERESFDQDSREEDPWDRL",
+    #        "MTARVRNRSSSRSYVLDFADLADGQREVLLPESRGNASEVDLPAGTTVNVTIDVTASGTGTLTARTPDGADVVSNEYELTVERDTDLTRVETESPQVAAGETATVTGTAENVGTVAGEREVTAYVDGE",
+    #        "MTAAGWREEGTPFARIARQLGRHVTSVRQAAGRVRQQMGLTSPDPADPPRSGPTPTIPIEQERA",
     seqChoice = gr.Radio(seqs, label="Generated sequences")
     btn.click(fn=update_protGPT2, inputs=[inp, length], outputs=seqChoice)
     gr.Markdown("## AlphaFold")
@@ -365,6 +365,6 @@ with proteindream:
     """
     )
     seqChoice.change(fn=update_seqs, inputs=seqChoice, outputs=chosenSeq)
-    btn2.click(fn=update, inputs=seqChoice, outputs=[mol, plot, meanpLDDT])
+    btn2.click(fn=update, inputs=chosenSeq, outputs=[mol, plot, meanpLDDT])
 
 proteindream.launch(share=False)