Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -5,7 +5,6 @@ import numpy as np
|
|
5 |
from PIL import Image
|
6 |
from einops import rearrange
|
7 |
import requests
|
8 |
-
import spaces
|
9 |
from diffusers.utils import load_image
|
10 |
from diffusers import FluxControlNetPipeline, FluxControlNetModel
|
11 |
from gradio_imageslider import ImageSlider
|
@@ -34,7 +33,6 @@ def preprocess_image(image, target_width, target_height):
|
|
34 |
image = image.resize((target_width, target_height), Image.LANCZOS)
|
35 |
return image
|
36 |
|
37 |
-
@spaces.GPU(duration=120)
|
38 |
def generate_image(prompt, control_image, control_mode, controlnet_conditioning_scale, num_steps, guidance, width, height, seed, random_seed):
|
39 |
if random_seed:
|
40 |
seed = np.random.randint(0, 10000)
|
@@ -70,7 +68,7 @@ interface = gr.Interface(
|
|
70 |
inputs=[
|
71 |
gr.Textbox(label="Prompt"),
|
72 |
gr.Image(type="pil", label="Control Image"),
|
73 |
-
gr.Dropdown(choices=[f"{i}: {name}" for i, name in CONTROL_MODES.items()], type="index", label="Control Mode",
|
74 |
gr.Slider(minimum=0.1, maximum=1.0, step=0.1, value=0.5, label="ControlNet Conditioning Scale"),
|
75 |
gr.Slider(step=1, minimum=1, maximum=64, value=24, label="Num Steps"),
|
76 |
gr.Slider(minimum=0.1, maximum=10, value=3.5, label="Guidance"),
|
|
|
5 |
from PIL import Image
|
6 |
from einops import rearrange
|
7 |
import requests
|
|
|
8 |
from diffusers.utils import load_image
|
9 |
from diffusers import FluxControlNetPipeline, FluxControlNetModel
|
10 |
from gradio_imageslider import ImageSlider
|
|
|
33 |
image = image.resize((target_width, target_height), Image.LANCZOS)
|
34 |
return image
|
35 |
|
|
|
36 |
def generate_image(prompt, control_image, control_mode, controlnet_conditioning_scale, num_steps, guidance, width, height, seed, random_seed):
|
37 |
if random_seed:
|
38 |
seed = np.random.randint(0, 10000)
|
|
|
68 |
inputs=[
|
69 |
gr.Textbox(label="Prompt"),
|
70 |
gr.Image(type="pil", label="Control Image"),
|
71 |
+
gr.Dropdown(choices=[f"{i}: {name}" for i, name in CONTROL_MODES.items()], type="index", label="Control Mode", value=0),
|
72 |
gr.Slider(minimum=0.1, maximum=1.0, step=0.1, value=0.5, label="ControlNet Conditioning Scale"),
|
73 |
gr.Slider(step=1, minimum=1, maximum=64, value=24, label="Num Steps"),
|
74 |
gr.Slider(minimum=0.1, maximum=10, value=3.5, label="Guidance"),
|