multimodalart HF Staff commited on
Commit
6a2dbc2
·
verified ·
1 Parent(s): 460c462

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -4,6 +4,7 @@ import os
4
  import random
5
  import numpy as np
6
  from PIL import Image
 
7
 
8
  # --- Model & Pipeline Imports ---
9
  from diffusers import QwenImageControlNetPipeline, QwenImageControlNetModel, FlowMatchEulerDiscreteScheduler
@@ -94,6 +95,7 @@ print("Loading complete.")
94
  # --- 3. Gradio UI and Generation Function ---
95
  MAX_SEED = np.iinfo(np.int32).max
96
 
 
97
  def generate(
98
  image,
99
  prompt,
 
4
  import random
5
  import numpy as np
6
  from PIL import Image
7
+ import spaces
8
 
9
  # --- Model & Pipeline Imports ---
10
  from diffusers import QwenImageControlNetPipeline, QwenImageControlNetModel, FlowMatchEulerDiscreteScheduler
 
95
  # --- 3. Gradio UI and Generation Function ---
96
  MAX_SEED = np.iinfo(np.int32).max
97
 
98
+ @spaces.GPU(duration=120)
99
  def generate(
100
  image,
101
  prompt,