Update app.py

app.py

@@ -7,10 +7,8 @@ from evodiff.generate import generate_oaardm, generate_d3pm
 from evodiff.generate_msa import generate_query_oadm_msa_simple
 
 import py3Dmol
-from colabfold.download import download_alphafold_params, default_data_dir
-from colabfold.utils import setup_logging
-from colabfold.batch import get_queries, run, set_model_type
-from colabfold.plot import plot_msa_v2
+from colabfold.download import download_alphafold_params
+from colabfold.batch import run
 
 def a3m_file(file):
     return "tmp.a3m"
@@ -18,7 +16,7 @@ def a3m_file(file):
 def predict_protein(sequence):
     download_alphafold_params("alphafold2_ptm", Path("."))
     results = run(
-        queries=[('evodiff_protein',sequence, None)],
+        queries=[('evodiff_protein', sequence, None)],
         result_dir='evodiff_protein',
         use_templates=False,
         num_relax=0,
@@ -33,7 +31,7 @@ def predict_protein(sequence):
         rank_by="auto",
         stop_at_score=float(100),
         zip_results=False,
-        user_agent="colabfold/google-colab-main",
+        user_agent="colabfold/google-colab-main"
     )
 
     return f"evodiff_protein/evodiff_protein_unrelaxed_rank_001_alphafold2_ptm_model_1_seed_000.pdb"
@@ -98,7 +96,7 @@ usg_app = gr.Interface(
             fn=make_uncond_seq,
             inputs=[
                 gr.Slider(10, 100, step=1, label = "Sequence Length"),
-                gr.Dropdown(["EvoDiff-Seq-OADM 38M", "EvoDiff-D3PM-Uniform 38M"], default="EvoDiff-Seq-OADM 38M", type="value", label = "Model")
+                gr.Dropdown(["EvoDiff-Seq-OADM 38M", "EvoDiff-D3PM-Uniform 38M"], value="EvoDiff-Seq-OADM 38M", type="value", label = "Model")
                 ],
             outputs=[
                 "text",
@@ -113,7 +111,7 @@ csg_app = gr.Interface(
             inputs=[
                 gr.Slider(10, 100, label = "Sequence Length"),
                 gr.File(file_types=["a3m"], label = "MSA File"),
-                gr.Dropdown(["EvoDiff-MSA"], default="EvoDiff-MSA", type="value", label = "Model")
+                gr.Dropdown(["EvoDiff-MSA"], value="EvoDiff-MSA", type="value", label = "Model")
                 ],
             outputs=[
                 "text",