multimodalart HF Staff commited on
Commit
ee824ed
·
verified ·
1 Parent(s): bff057a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -32,7 +32,7 @@ if os.path.isdir(_SRC_DIR):
32
  print(f"[setup] Added {_SRC_DIR} to sys.path")
33
 
34
  import gradio as gr
35
- #import spaces
36
  import torch
37
 
38
  from diffusers import (
@@ -62,7 +62,7 @@ pipe.to("cuda")
62
  # ---------------------------------------------------------------------------
63
  # Generation (decorated for ZeroGPU)
64
  # ---------------------------------------------------------------------------
65
- # @spaces.GPU(duration=300)
66
  def generate_video(
67
  mode: str,
68
  prompt: str,
 
32
  print(f"[setup] Added {_SRC_DIR} to sys.path")
33
 
34
  import gradio as gr
35
+ import spaces
36
  import torch
37
 
38
  from diffusers import (
 
62
  # ---------------------------------------------------------------------------
63
  # Generation (decorated for ZeroGPU)
64
  # ---------------------------------------------------------------------------
65
+ @spaces.GPU(duration=300)
66
  def generate_video(
67
  mode: str,
68
  prompt: str,