Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
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,
|