Manjushri commited on
Commit
52cd608
1 Parent(s): 892d658

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -4
app.py CHANGED
@@ -15,16 +15,18 @@ def resize(value,img):
15
  img = img.resize((value,value))
16
  return img
17
 
18
- def infer(source_img, prompt, negative_prompt, guide, steps, seed):
19
  generator = torch.Generator(device).manual_seed(seed)
20
  imageio.imwrite("data.png", source_img)
21
  src = resize(768, 'data.png')
22
  image = pipe(prompt, negative_prompt=negative_prompt, image=src, strength=1, guidance_scale=guide, num_inference_steps=steps).images[0]
23
  return image
24
 
25
- gr.Interface(fn=infer, inputs=[gr.Image(type='numpy', source='canvas'), gr.Textbox(label = 'Prompt Input Text. 77 Token (Keyword or Symbol) Maximum'), gr.Textbox(label='What you Do Not want the AI to generate.'),
26
- gr.Slider(5, 15, value = 10, label = 'Guidance Scale'),
27
- gr.Slider(25, 50, value = 25, step = 1, label = 'Number of Iterations'),
 
 
28
  gr.Slider(label = "Seed", minimum = 0, maximum = 987654321987654321, step = 1, randomize = True)],
29
  outputs='image',
30
  title = "Stable Diffusion XL 1.0 Doodle to Image CPU",
 
15
  img = img.resize((value,value))
16
  return img
17
 
18
+ def infer(source_img, prompt, negative_prompt, steps, seed):
19
  generator = torch.Generator(device).manual_seed(seed)
20
  imageio.imwrite("data.png", source_img)
21
  src = resize(768, 'data.png')
22
  image = pipe(prompt, negative_prompt=negative_prompt, image=src, strength=1, guidance_scale=guide, num_inference_steps=steps).images[0]
23
  return image
24
 
25
+ gr.Interface(fn=infer, inputs=[gr.Image(type='numpy', source='canvas'),
26
+ gr.Textbox(label = 'Prompt Input Text. 77 Token (Keyword or Symbol) Maximum'),
27
+ gr.Textbox(label='What you Do Not want the AI to generate.'),
28
+ #gr.Slider(5, 15, value = 10, label = 'Guidance Scale'),
29
+ gr.Slider(1, 5, value = 2, step = 1, label = 'Number of Iterations'),
30
  gr.Slider(label = "Seed", minimum = 0, maximum = 987654321987654321, step = 1, randomize = True)],
31
  outputs='image',
32
  title = "Stable Diffusion XL 1.0 Doodle to Image CPU",