Update app.py

app.py

@@ -6,6 +6,30 @@ from gradio_molecule3d import Molecule3D
 from gradio_cofoldinginput import CofoldingInput
 
 import os
+import urllib.request
+
+CCD_URL = "https://huggingface.co/boltz-community/boltz-1/resolve/main/ccd.pkl"
+MODEL_URL = "https://huggingface.co/boltz-community/boltz-1/resolve/main/boltz1.ckpt"
+
+cache = "~/.boltz"
+ccd = cache / "ccd.pkl"
+if not ccd.exists():
+    click.echo(
+        f"Downloading the CCD dictionary to {ccd}. You may "
+        "change the cache directory with the --cache flag."
+    )
+    urllib.request.urlretrieve(CCD_URL, str(ccd))  # noqa: S310
+
+# Download model
+model = cache / "boltz1.ckpt"
+if not model.exists():
+    click.echo(
+        f"Downloading the model weights to {model}. You may "
+        "change the cache directory with the --cache flag."
+    )
+    urllib.request.urlretrieve(MODEL_URL, str(model))  # noqa: S310
+
+
 
 @spaces.GPU(duration=120)
 def predict(jobname, inputs, recycling_steps, sampling_steps, diffusion_samples):