add soluble model weights

ProteinMPNN/soluble_model_weights/v_48_010.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:39059256e354a5ade71a6f14e84ca418bc27c15e65ac0cdf7ba6c215b4418775
+size 6681301

ProteinMPNN/soluble_model_weights/v_48_020.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:e491ffe7e25224d031daf4a01ba9da880e399eb63e094d1772d99467cfbbab5c
+size 6681301

app.py

@@ -236,15 +236,10 @@ def run_alphafold(sequences, num_recycles):
     return plddts, paes, LEN
 
 
-if os.path.exists("/home/duerr/phd/08_Code/ProteinMPNN"):
-    path_to_model_weights = "/home/duerr/phd/08_Code/ProteinMPNN/ProteinMPNN/vanilla_proteinmpnn/vanilla_model_weights"
-else:
-    path_to_model_weights = (
-        "/home/user/app/ProteinMPNN/vanilla_proteinmpnn/vanilla_model_weights"
-    )
 
 
-def setup_proteinmpnn(model_name="v_48_020", backbone_noise=0.00):
+
+def setup_proteinmpnn(model_name="vanilla—v_48_020", backbone_noise=0.00):
     from protein_mpnn_utils import (
         loss_nll,
         loss_smoothed,
@@ -266,6 +261,17 @@ def setup_proteinmpnn(model_name="v_48_020", backbone_noise=0.00):
     # Standard deviation of Gaussian noise to add to backbone atoms
     hidden_dim = 128
     num_layers = 3
+
+    model, model_name = model_name.split("—")
+    if os.path.exists("/home/duerr"):
+        dir = "/home/duerr/phd/08_Code/ProteinMPNN"
+    else:
+        dir = "/home/user/app"
+
+    path_to_model_weights = (
+        f"{dir}/ProteinMPNN/{model}_model_weights"
+    )
+
     model_folder_path = path_to_model_weights
     if model_folder_path[-1] != "/":
         model_folder_path = model_folder_path + "/"
@@ -1265,13 +1271,15 @@ with proteinMPNN:
             with gr.Row():
                 model_name = gr.Dropdown(
                     choices=[
-                        "v_48_002",
-                        "v_48_010",
-                        "v_48_020",
-                        "v_48_030",
+                        "vanilla—v_48_002",
+                        "vanilla—v_48_010",
+                        "vanilla—v_48_020",
+                        "vanilla—v_48_030",
+                        "soluble—v_48_010",
+                        "soluble—v_48_020",
                     ],
                     label="Model",
-                    value="v_48_020",
+                    value="vanilla—v_48_020",
                 )
                 backbone_noise = gr.Dropdown(
                     choices=[0, 0.02, 0.10, 0.20, 0.30], label="Backbone noise", value=0