Omnibus commited on
Commit
b798e60
·
1 Parent(s): 354039e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -7
app.py CHANGED
@@ -1,6 +1,7 @@
1
  import gradio as gr
2
  import os
3
  import sys
 
4
  os.system("mkdir hotshotxl")
5
  os.system("cd hotshotxl")
6
  #os.system("mkdir -p datasets/DeepFashion")
@@ -12,18 +13,14 @@ os.system("pip install -r requirements.txt")
12
  import inference as inf
13
  def infer(inp):
14
 
15
- os.system("""python Hotshot-XL/inference.py \
16
- --prompt="a bulldog in the captains chair of a spaceship, hd, high quality" \
17
- --output="output.gif"
18
- """)
19
-
20
 
21
  #output = inf("a cat running through a field")
22
- return "output.gif"
23
 
24
  with gr.Blocks() as app:
25
  prompt = gr.Textbox()
26
- gif_out = gr.Video()
27
  btn = gr.Button()
28
  btn.click(infer,prompt,gif_out)
29
  app.launch()
 
1
  import gradio as gr
2
  import os
3
  import sys
4
+ from huggingface_hub import scan_cache_dir
5
  os.system("mkdir hotshotxl")
6
  os.system("cd hotshotxl")
7
  #os.system("mkdir -p datasets/DeepFashion")
 
13
  import inference as inf
14
  def infer(inp):
15
 
16
+ #hf_cache_info = scan_cache_dir()
 
 
 
 
17
 
18
  #output = inf("a cat running through a field")
19
+ return scan_cache_dir()
20
 
21
  with gr.Blocks() as app:
22
  prompt = gr.Textbox()
23
+ gif_out = gr.JSON()
24
  btn = gr.Button()
25
  btn.click(infer,prompt,gif_out)
26
  app.launch()