move model ladoing back

app.py

@@ -393,21 +393,6 @@ 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 = 'cuda' if torch.cuda.is_available() else 'cpu'
-
-# Load models
-models = [
-    DDGPPIformer.load_from_checkpoint(
-        PPIFORMER_WEIGHTS_DIR / f'ddg_regression/{i}.ckpt',
-        map_location=torch.device('cpu')
-    ).eval()
-    for i in range(3)
-]
-models = [model.to(device) for model in models]
-
-
 app = gr.Blocks(theme=gr.themes.Default(primary_hue="green", secondary_hue="pink"))
 with app:
 
@@ -502,6 +487,20 @@ with app:
     # Download weights from Zenodo
     download_weights()
 
+    # Set device
+    # device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
+    device = 'cuda' if torch.cuda.is_available() else 'cpu'
+
+    # Load models
+    models = [
+        DDGPPIformer.load_from_checkpoint(
+            PPIFORMER_WEIGHTS_DIR / f'ddg_regression/{i}.ckpt',
+            map_location=torch.device('cpu')
+        ).eval()
+        for i in range(3)
+    ]
+    models = [model.to(device) for model in models]
+
     # Create temporary directory for storing downloaded PDBs and extracted PPIs
     temp_dir_obj = tempfile.TemporaryDirectory()
     temp_dir = Path(temp_dir_obj.name)