Move out model loading from the app

app.py

@@ -393,6 +393,20 @@ def update_plot(dropdown, dropdown_choices_to_plot_args):
     return plot_3dmol(*dropdown_choices_to_plot_args[dropdown])
 
 
+# Set device
+device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
+# device = torch.device('cuda')
+
+# Load models
+models = [
+    DDGPPIformer.load_from_checkpoint(
+        PPIFORMER_WEIGHTS_DIR / f'ddg_regression/{i}.ckpt',
+        map_location=device
+    ).eval()
+    for i in range(3)
+]
+
+
 app = gr.Blocks(theme=gr.themes.Default(primary_hue="green", secondary_hue="pink"))
 with app:
 
@@ -487,19 +501,6 @@ with app:
     # Download weights from Zenodo
     download_weights()
 
-    # Set device
-    device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
-    # device = torch.device('cuda')
-
-    # Load models
-    models = [
-        DDGPPIformer.load_from_checkpoint(
-            PPIFORMER_WEIGHTS_DIR / f'ddg_regression/{i}.ckpt',
-            map_location=device
-        ).eval()
-        for i in range(3)
-    ]
-
     # Create temporary directory for storing downloaded PDBs and extracted PPIs
     temp_dir_obj = tempfile.TemporaryDirectory()
     temp_dir = Path(temp_dir_obj.name)