Deadmon commited on
Commit
c35a37d
·
verified ·
1 Parent(s): 982ad81

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -3
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", default=0),
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"),