Update app.py

app.py

@@ -53,24 +53,24 @@ def make_inpainted_idrs(sequence, start_idx, end_idx, model_type):
 
     return generated_idr_output
     
-def make_scaffold_motifs(pdb_code, start_idx, end_idx, scaffold_length, model_type):
-    if model_type == "EvoDiff-Seq":
-        checkpoint = OA_DM_38M()
-        model, collater, tokenizer, scheme = checkpoint
-        data_top_dir = '/home/user/.cache/huggingface/datasets/'
-        os.makedirs(data_top_dir, exist_ok=True)
-        # print("Folders in User Cache Directory:", os.listdir("/home/user/.cache"))
-        start_idx = list(map(int, start_idx.strip('][').split(',')))
-        end_idx = list(map(int, end_idx.strip('][').split(',')))
-        generated_sequence, new_start_idx, new_end_idx = generate_scaffold(model, pdb_code, start_idx, end_idx, scaffold_length, data_top_dir, tokenizer, device=get_device())
-
-        generated_scaffold_output = {
-            "generated_sequence": generated_sequence,
-            "new_start_index": new_start_idx,
-            "new_end_index": new_end_idx
-        }
-
-    return generated_scaffold_output
+# def make_scaffold_motifs(pdb_code, start_idx, end_idx, scaffold_length, model_type):
+#     if model_type == "EvoDiff-Seq":
+#         checkpoint = OA_DM_38M()
+#         model, collater, tokenizer, scheme = checkpoint
+#         data_top_dir = '/home/user/.cache/huggingface/datasets/'
+#         os.makedirs(data_top_dir, exist_ok=True)
+#         # print("Folders in User Cache Directory:", os.listdir("/home/user/.cache"))
+#         start_idx = list(map(int, start_idx.strip('][').split(',')))
+#         end_idx = list(map(int, end_idx.strip('][').split(',')))
+#         generated_sequence, new_start_idx, new_end_idx = generate_scaffold(model, pdb_code, start_idx, end_idx, scaffold_length, data_top_dir, tokenizer, device=get_device())
+
+#         generated_scaffold_output = {
+#             "generated_sequence": generated_sequence,
+#             "new_start_index": new_start_idx,
+#             "new_end_index": new_end_idx
+#         }
+
+#     return generated_scaffold_output
 
 usg_app = gr.Interface(
             fn=make_uncond_seq,
@@ -111,19 +111,19 @@ idr_app = gr.Interface(
             description="Inpainting a new region inside a given sequence using the `EvoDiff-Seq` model."
             )
 
-scaffold_app = gr.Interface(
-            fn=make_scaffold_motifs,
-            inputs=[
-                gr.Textbox(value="1prw", label = "PDB Code"),
-                gr.Textbox(value="[15, 51]", label = "Start Index (as list)"),
-                gr.Textbox(value="[34, 70]", label = "End Index (as list)"),
-                gr.Number(value=75, precision=0, label = "Scaffold Length"),
-                gr.Dropdown(["EvoDiff-Seq", "EvoDiff-MSA"], value="EvoDiff-Seq", type="value", label = "Model")
-                ],
-            outputs=["text"],
-            title = "Scaffolding functional motifs",
-            description="Scaffolding a new functional motif inside a given PDB structure using the `EvoDiff-Seq` model."
-            )
+# scaffold_app = gr.Interface(
+#             fn=make_scaffold_motifs,
+#             inputs=[
+#                 gr.Textbox(value="1prw", label = "PDB Code"),
+#                 gr.Textbox(value="[15, 51]", label = "Start Index (as list)"),
+#                 gr.Textbox(value="[34, 70]", label = "End Index (as list)"),
+#                 gr.Number(value=75, precision=0, label = "Scaffold Length"),
+#                 gr.Dropdown(["EvoDiff-Seq", "EvoDiff-MSA"], value="EvoDiff-Seq", type="value", label = "Model")
+#                 ],
+#             outputs=["text"],
+#             title = "Scaffolding functional motifs",
+#             description="Scaffolding a new functional motif inside a given PDB structure using the `EvoDiff-Seq` model."
+#             )
 
 with gr.Blocks() as edapp:
     with gr.Row():
@@ -132,7 +132,7 @@ with gr.Blocks() as edapp:
             # EvoDiff
             ## Generation of protein sequences and evolutionary alignments via discrete diffusion models
 
-            Created By: Microsoft Research [Sarah Alamdari, Nitya Thakkar, Rianne van den Berg, Alex X. Lu, Nicolo Fusi, ProfileAva P. Amini, and Kevin K. Yang]
+            Created By: Microsoft Research [Sarah Alamdari, Nitya Thakkar, Rianne van den Berg, Alex X. Lu, Nicolo Fusi, Ava P. Amini, and Kevin K. Yang]
             
             Spaces App By: Tuple, The Cloud Genomics Company [Colby T. Ford]
             """
@@ -141,14 +141,14 @@ with gr.Blocks() as edapp:
         gr.TabbedInterface([
             usg_app,
             csg_app,
-            idr_app,
-            scaffold_app
+            idr_app#,
+            # scaffold_app
             ],
             [
                 "Unconditional sequence generation",
                 "Conditional generation",
-                "Inpainting IDRs",
-                "Scaffolding functional motifs"
+                "Inpainting IDRs"#,
+                # "Scaffolding functional motifs"
             ])