simonduerr commited on
Commit
8704b30
1 Parent(s): f07faaf

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -7
app.py CHANGED
@@ -12,22 +12,20 @@ CCD_URL = "https://huggingface.co/boltz-community/boltz-1/resolve/main/ccd.pkl"
12
  MODEL_URL = "https://huggingface.co/boltz-community/boltz-1/resolve/main/boltz1.ckpt"
13
 
14
  cache = "~/.boltz"
15
- ccd = cache / "ccd.pkl"
16
  if not ccd.exists():
17
  print(
18
  f"Downloading the CCD dictionary to {ccd}. You may "
19
- "change the cache directory with the --cache flag."
20
  )
21
- urllib.request.urlretrieve(CCD_URL, str(ccd)) # noqa: S310
22
 
23
  # Download model
24
- model = cache / "boltz1.ckpt"
25
  if not model.exists():
26
  print(
27
- f"Downloading the model weights to {model}. You may "
28
- "change the cache directory with the --cache flag."
29
  )
30
- urllib.request.urlretrieve(MODEL_URL, str(model)) # noqa: S310
31
 
32
 
33
 
 
12
  MODEL_URL = "https://huggingface.co/boltz-community/boltz-1/resolve/main/boltz1.ckpt"
13
 
14
  cache = "~/.boltz"
15
+ ccd = f"{cache}/ccd.pkl"
16
  if not ccd.exists():
17
  print(
18
  f"Downloading the CCD dictionary to {ccd}. You may "
 
19
  )
20
+ urllib.request.urlretrieve(CCD_URL, str(ccd))
21
 
22
  # Download model
23
+ model =f"{cache}/boltz1.ckpt"
24
  if not model.exists():
25
  print(
26
+ f"Downloading the model weights to {model}"
 
27
  )
28
+ urllib.request.urlretrieve(MODEL_URL, str(model))
29
 
30
 
31