gabboud commited on
Commit
605b2c5
·
1 Parent(s): 8b91ed4

second try at consistent weights

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -13,11 +13,11 @@ from rfd3.engine import RFD3InferenceConfig, RFD3InferenceEngine
13
  # Download model weights (skips already-downloaded models automatically)
14
  # In total, ~6GB (3GB for RFD3, 3GB for RF3, <100MB for MPNN); may take a few minutes depending on your connection speed
15
 
16
- cmd = "foundry install rfd3 ligandmpnn rf3"
17
- print("Global PATH:", os.environ.get("PATH"))
18
 
19
- #os.environ["HF_HOME"] = "/data" # HF Spaces persistent dir
20
- #os.environ["FOUNDRY_CHECKPOINT_DIRS"] = "/data/checkpoints"
 
 
21
 
22
  result = subprocess.run(cmd, shell=True, capture_output=True, text=True)
23
  if result.returncode == 0:
 
13
  # Download model weights (skips already-downloaded models automatically)
14
  # In total, ~6GB (3GB for RFD3, 3GB for RF3, <100MB for MPNN); may take a few minutes depending on your connection speed
15
 
 
 
16
 
17
+ cmd = f"foundry install rfd3 ligandmpnn rf3 --checkpoint-dir /data/checkpoints"
18
+ print("Global PATH:", os.environ.get("PATH"))
19
+ os.environ["HF_HOME"] = "/data" # HF Spaces persistent dir
20
+ os.environ["FOUNDRY_CHECKPOINT_DIRS"] = "/data/checkpoints"
21
 
22
  result = subprocess.run(cmd, shell=True, capture_output=True, text=True)
23
  if result.returncode == 0: