tedlasai commited on
Commit
bbfa81d
·
1 Parent(s): f3bf944

added gpu duration warning

Browse files
Files changed (2) hide show
  1. README.md +4 -4
  2. gradio/app.py +2 -2
README.md CHANGED
@@ -1,8 +1,8 @@
1
  ---
2
- title: Blur2vid
3
- emoji: 📈
4
- colorFrom: pink
5
- colorTo: pink
6
  sdk: gradio
7
  sdk_version: 5.49.1
8
  app_file: gradio/app.py
 
1
  ---
2
+ title: blur2vid
3
+ emoji: 🎥
4
+ colorFrom: red
5
+ colorTo: indigo
6
  sdk: gradio
7
  sdk_version: 5.49.1
8
  app_file: gradio/app.py
gradio/app.py CHANGED
@@ -27,7 +27,7 @@ OUTPUT_DIR = Path("/tmp/generated_videos")
27
  OUTPUT_DIR.mkdir(parents=True, exist_ok=True)
28
 
29
 
30
- @spaces.GPU(timeout=300, duration=200)
31
  def generate_video_from_image(image: Image.Image, interval_key: str, num_inference_steps: int) -> str:
32
  """
33
  Wrapper for Gradio. Takes an image and returns a video path.
@@ -96,7 +96,7 @@ with gr.Blocks(css="footer {visibility: hidden}") as demo:
96
  maximum=50,
97
  step=1,
98
  value=20,
99
- info="More steps = better quality but slower",
100
  )
101
 
102
  generate_btn = gr.Button("Generate video", variant="primary")
 
27
  OUTPUT_DIR.mkdir(parents=True, exist_ok=True)
28
 
29
 
30
+ @spaces.GPU(timeout=300, duration=120)
31
  def generate_video_from_image(image: Image.Image, interval_key: str, num_inference_steps: int) -> str:
32
  """
33
  Wrapper for Gradio. Takes an image and returns a video path.
 
96
  maximum=50,
97
  step=1,
98
  value=20,
99
+ info="More steps = better quality but slower (Warning: too many steps may time out ZeroGPU)",
100
  )
101
 
102
  generate_btn = gr.Button("Generate video", variant="primary")