Update rosettafold_pymol.py

rosettafold_pymol.py

@@ -142,7 +142,7 @@ def query_rosettafold2(
         url = "http://localhost:7860"
     except requests.exceptions.ConnectionError:
         url = "https://simonduerr-rosettafold2.hf.space"
-
+    print(f'querying {url}') 
     response = requests.post(
         url + "/run/rosettafold2/",
         json={
@@ -151,16 +151,16 @@ def query_rosettafold2(
                 jobname,  # str  in 'jobname' Textbox component
                 sym,  # str  in 'sym' Textbox component
                 order,  # int | float (numeric value between 1 and 12) in 'order' Slider component
-                "diag",  # str (Option from: ['diag', 'repeat', 'default']) in 'msa_concat_mode' Dropdown component
-                "single_sequence",  # str (Option from: ['mmseqs2', 'single_sequence', 'custom_a3m']) in 'msa_method' Dropdown component
-                "unpaired_paired",  # str (Option from: ['unpaired_paired', 'paired', 'unpaired']) in 'pair_mode' Dropdown component
-                False,  # bool  in 'collapse_identical' Checkbox component
-                6,  # int (Option from: ['0', '1', '3', '6', '12', '24']) in 'num_recycles' Dropdown component
-                False,  # bool  in 'use_mlm' Checkbox component
-                False,  # bool  in 'use_dropout' Checkbox component
-                16,  # int (Option from: ['16', '32', '64', '128', '256', '512']) in 'max_msa' Dropdown component
-                0,  # int  in 'random_seed' Textbox component
-                1,  # int (Option from: ['1', '2', '4', '8', '16', '32']) in 'num_models' Dropdown component
+                msa_concat_mode,  # str (Option from: ['diag', 'repeat', 'default']) in 'msa_concat_mode' Dropdown component
+                msa_method,  # str (Option from: ['mmseqs2', 'single_sequence', 'custom_a3m']) in 'msa_method' Dropdown component
+                pair_mode,  # str (Option from: ['unpaired_paired', 'paired', 'unpaired']) in 'pair_mode' Dropdown component
+                collapse_identical,  # bool  in 'collapse_identical' Checkbox component
+                num_recycles,  # int (Option from: ['0', '1', '3', '6', '12', '24']) in 'num_recycles' Dropdown component
+                use_mlm,  # bool  in 'use_mlm' Checkbox component
+                use_dropout,  # bool  in 'use_dropout' Checkbox component
+                max_msa,  # int (Option from: ['16', '32', '64', '128', '256', '512']) in 'max_msa' Dropdown component
+                random_seed,  # int  in 'random_seed' Textbox component
+                num_models,  # int (Option from: ['1', '2', '4', '8', '16', '32']) in 'num_models' Dropdown component
             ]
         },
     ).json()