Stop downloading models

app.py

@@ -5,7 +5,6 @@ import gradio as gr
 import numpy as np
 import os
 import torch
-import subprocess
 import output
 
 from rdkit import Chem
@@ -53,24 +52,15 @@ args = parser.parse_args()
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 print(f'Device: {device}')
 os.makedirs("results", exist_ok=True)
-os.makedirs("models", exist_ok=True)
 
 size_gnn_path = 'models/geom_size_gnn.ckpt'
-if not os.path.exists(size_gnn_path):
-    print('Downloading SizeGNN model...')
-    link = 'https://zenodo.org/record/7121300/files/geom_size_gnn.ckpt?download=1'
-    subprocess.run(f'wget {link} -O {size_gnn_path}', shell=True)
 size_nn = SizeClassifier.load_from_checkpoint('models/geom_size_gnn.ckpt', map_location=device).eval().to(device)
 print('Loaded SizeGNN model')
 
 
 diffusion_models = {}
 for model_name, metadata in MODELS_METADATA.items():
-    link = metadata['link']
     diffusion_path = metadata['path']
-    if not os.path.exists(diffusion_path):
-        print(f'Downloading {model_name}...')
-        subprocess.run(f'wget {link} -O {diffusion_path}', shell=True)
     diffusion_models[model_name] = DDPM.load_from_checkpoint(diffusion_path, map_location=device).eval().to(device)
     print(f'Loaded model {model_name}')