apolinario commited on
Commit
8c6eab4
·
1 Parent(s): a9765cd

Update due to Gradio bug

Browse files
Files changed (1) hide show
  1. app.py +6 -7
app.py CHANGED
@@ -46,7 +46,7 @@ model = load_model_from_config(config, f"txt2img-f8-large.ckpt")
46
  device = torch.device("cuda") if torch.cuda.is_available() else torch.device("cpu")
47
  model = model.to(device)
48
 
49
- def run(prompt, steps, width, height, images, scale, eta):
50
  if images == 6:
51
  images = 3
52
  n_iter = 2
@@ -56,7 +56,7 @@ def run(prompt, steps, width, height, images, scale, eta):
56
  prompt = prompt,
57
  outdir='latent-diffusion/outputs',
58
  ddim_steps = int(steps),
59
- ddim_eta = eta,
60
  n_iter = n_iter,
61
  W=int(width),
62
  H=int(height),
@@ -126,12 +126,11 @@ css = ".output-image{height: 528px !important} .output-carousel .output-image{he
126
  iface = gr.Interface(fn=run, inputs=[
127
  gr.inputs.Textbox(label="Prompt",default="A drawing of a cute dog with a funny hat"),
128
  gr.inputs.Slider(label="Steps - more steps can increase quality but will take longer to generate",default=50,maximum=250,minimum=1,step=1),
129
- gr.inputs.Slider(label="Width", minimum=64, maximum=256, default=256, step=64),
130
- gr.inputs.Slider(label="Height", minimum=64, maximum=256, default=256, step=64),
131
  gr.inputs.Slider(label="Images - How many images you wish to generate", default=4, step=2, minimum=2, maximum=6),
132
- gr.inputs.Slider(label="Diversity scale - How different from one another you wish the images to be",default=5.0, minimum=1),
133
- gr.inputs.Slider(label="ETA - between 0 and 1. Lower values can provide better quality, higher values can be more diverse",default=0.0,minimum=0.0, maximum=1.0,step=0.1),
134
-
135
  ],
136
  outputs=[image,gr.outputs.Carousel(label="Individual images",components=["image"])],
137
  css=css,
 
46
  device = torch.device("cuda") if torch.cuda.is_available() else torch.device("cpu")
47
  model = model.to(device)
48
 
49
+ def run(prompt, steps, width, height, images, scale):
50
  if images == 6:
51
  images = 3
52
  n_iter = 2
 
56
  prompt = prompt,
57
  outdir='latent-diffusion/outputs',
58
  ddim_steps = int(steps),
59
+ ddim_eta = 0,
60
  n_iter = n_iter,
61
  W=int(width),
62
  H=int(height),
 
126
  iface = gr.Interface(fn=run, inputs=[
127
  gr.inputs.Textbox(label="Prompt",default="A drawing of a cute dog with a funny hat"),
128
  gr.inputs.Slider(label="Steps - more steps can increase quality but will take longer to generate",default=50,maximum=250,minimum=1,step=1),
129
+ gr.inputs.Radio(label="Width", choices=[32,64,128,256,384],default=256),
130
+ gr.inputs.Radio(label="Height", choices=[32,64,128,256,384],default=256),
131
  gr.inputs.Slider(label="Images - How many images you wish to generate", default=4, step=2, minimum=2, maximum=6),
132
+ gr.inputs.Slider(label="Diversity scale - How different from one another you wish the images to be",default=5.0, minimum=1.0, maximum=50),
133
+ #gr.inputs.Slider(label="ETA - between 0 and 1. Lower values can provide better quality, higher values can be more diverse",default=0.0,minimum=0.0, maximum=1.0,step=0.1),
 
134
  ],
135
  outputs=[image,gr.outputs.Carousel(label="Individual images",components=["image"])],
136
  css=css,