download models

app.py

@@ -53,11 +53,19 @@ allowpaymentrequest="" frameborder="0" srcdoc='{html}'></iframe>"""
 
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 os.makedirs("results", exist_ok=True)
-print('Created results directory')
+os.makedirs("models", exist_ok=True)
 
+subprocess.run(
+    'wget https://zenodo.org/record/7121300/files/geom_size_gnn.ckpt?download=1 -O models/geom_size_gnn.ckpt',
+    shell=True
+)
 size_nn = SizeClassifier.load_from_checkpoint('models/geom_size_gnn.ckpt', map_location=device).eval().to(device)
 print('Loaded SizeGNN model')
 
+subprocess.run(
+    'wget https://zenodo.org/record/7121300/files/geom_difflinker.ckpt?download=1 -O models/geom_difflinker.ckpt',
+    shell=True
+)
 ddpm = DDPM.load_from_checkpoint('models/geom_difflinker.ckpt', map_location=device).eval().to(device)
 print('Loaded diffusion model')