Manjushri commited on
Commit
052883d
·
1 Parent(s): a3a76c8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -3
app.py CHANGED
@@ -1,6 +1,5 @@
1
 
2
  import gradio as gr
3
- #import cv2
4
  import torch
5
  import numpy as np
6
  from PIL import Image
@@ -16,6 +15,6 @@ pipe = pipe.to(device)
16
 
17
  def genie (prompt, scale, steps, seed):
18
  generator = torch.Generator(device=device).manual_seed(seed)
19
- images = pipe(prompt, width=768, height=768, num_inference_steps=steps, guidance_scale=scale, num_images_per_prompt=1, generator=generator).images
20
  return images[0]
21
- gr.Interface(fn=genie, inputs=['text', gr.Slider(1, 10, 20), gr.Slider(), gr.Slider(maximum=987654321)], outputs='image').launch(debug=True)
 
1
 
2
  import gradio as gr
 
3
  import torch
4
  import numpy as np
5
  from PIL import Image
 
15
 
16
  def genie (prompt, scale, steps, seed):
17
  generator = torch.Generator(device=device).manual_seed(seed)
18
+ images = pipe(prompt, width=512, height=512, num_inference_steps=steps, guidance_scale=scale, num_images_per_prompt=1, generator=generator).images
19
  return images[0]
20
+ gr.Interface(fn=genie, inputs=['text', gr.Slider(1, 10, 20), gr.Slider(maximum=50), gr.Slider(maximum=987654321)], outputs='image').launch(debug=True)