tree3po commited on
Commit
b2615f7
1 Parent(s): 64e9db8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +8 -2
app.py CHANGED
@@ -2,11 +2,17 @@ import os
2
  #import huggingface_hub
3
  #from huggingface_hub import snapshot_download
4
  #import open_oasis_master as oa
5
-
6
  os.system("huggingface-cli login --token $HF_TOKEN --add-to-git-credential")
7
  os.system("huggingface-cli download Etched/oasis-500m oasis500m.pt --local-dir ./") # DiT checkpoint
8
  os.system("huggingface-cli download Etched/oasis-500m vit-l-20.pt --local-dir ./")
9
- os.system("python ./open_oasis_master/generate.py")
 
 
 
 
 
 
10
  '''os.system("git clone https://github.com/etched-ai/open-oasis.git")
11
  os.system("cd open-oasis")
12
  os.system("pip install -r open-oasis/requirements.txt")
 
2
  #import huggingface_hub
3
  #from huggingface_hub import snapshot_download
4
  #import open_oasis_master as oa
5
+ from open_oasis_master import generate
6
  os.system("huggingface-cli login --token $HF_TOKEN --add-to-git-credential")
7
  os.system("huggingface-cli download Etched/oasis-500m oasis500m.pt --local-dir ./") # DiT checkpoint
8
  os.system("huggingface-cli download Etched/oasis-500m vit-l-20.pt --local-dir ./")
9
+ #os.system("python ./open_oasis_master/generate.py")
10
+
11
+ with gr.Blocks() as app:
12
+ vid = gr.Video()
13
+ app.load(generate.run_mod(),None,vid)
14
+ app.launch()
15
+
16
  '''os.system("git clone https://github.com/etched-ai/open-oasis.git")
17
  os.system("cd open-oasis")
18
  os.system("pip install -r open-oasis/requirements.txt")