prithivMLmods commited on
Commit
3cd5fe0
·
verified ·
1 Parent(s): 481a54c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -42,7 +42,7 @@ aspect_ratios = {
42
  }
43
 
44
  # Generation function for Qwen/Qwen-Image
45
- @spaces.GPU
46
  def generate_qwen(
47
  prompt: str,
48
  negative_prompt: str = "",
@@ -89,7 +89,7 @@ def generate_qwen(
89
  return image_paths, seed, f"{duration:.2f}", zip_path
90
 
91
  # Wrapper function to handle UI logic
92
- @spaces.GPU
93
  def generate(
94
  prompt: str,
95
  negative_prompt: str,
 
42
  }
43
 
44
  # Generation function for Qwen/Qwen-Image
45
+ @spaces.GPU(duration=120)
46
  def generate_qwen(
47
  prompt: str,
48
  negative_prompt: str = "",
 
89
  return image_paths, seed, f"{duration:.2f}", zip_path
90
 
91
  # Wrapper function to handle UI logic
92
+ @spaces.GPU(duration=120)
93
  def generate(
94
  prompt: str,
95
  negative_prompt: str,