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