tree3po commited on
Commit
d9bccac
·
verified ·
1 Parent(s): 4ae1148

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -2
app.py CHANGED
@@ -1,6 +1,11 @@
1
  import os
 
2
  os.system("git clone https://github.com/etched-ai/open-oasis.git")
3
  os.system("cd open-oasis")
4
  os.system("pip install -r open-oasis/requirements.txt")
5
- os.system("huggingface-cli download Etched/oasis-500m oasis500m.pt") # DiT checkpoint
6
- os.system("huggingface-cli download Etched/oasis-500m vit-l-20.pt")
 
 
 
 
 
1
  import os
2
+ import huggingface_hub
3
  os.system("git clone https://github.com/etched-ai/open-oasis.git")
4
  os.system("cd open-oasis")
5
  os.system("pip install -r open-oasis/requirements.txt")
6
+ #os.system("huggingface-cli download Etched/oasis-500m oasis500m.pt") # DiT checkpoint
7
+ #os.system("huggingface-cli download Etched/oasis-500m vit-l-20.pt")
8
+ from huggingface_hub import hf_hub_download
9
+ oas = hf_hub_download(repo_id="Etched/oasis-500m", filename="oasis500m.pt")
10
+ vit = hf_hub_download(repo_id="Etched/oasis-500m", filename="vit-l-20.pt")
11
+ print(oas,vit)