Manjushri commited on
Commit
ed90fbc
1 Parent(s): 25e7c41

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -46,7 +46,7 @@ def genie (prompt, negative_prompt, scale, steps, seed, upscaler):
46
  pipe.enable_xformers_memory_efficient_attention()
47
  image = pipe(prompt=prompt, image=int_image).images[0]
48
  torch.cuda.empty_cache()
49
- return image
50
 
51
  gr.Interface(fn=genie, inputs=[gr.Textbox(label='What you want the AI to generate. 77 Token Limit.'),
52
  gr.Textbox(label='What you Do Not want the AI to generate.'),
 
46
  pipe.enable_xformers_memory_efficient_attention()
47
  image = pipe(prompt=prompt, image=int_image).images[0]
48
  torch.cuda.empty_cache()
49
+ return (image, image)
50
 
51
  gr.Interface(fn=genie, inputs=[gr.Textbox(label='What you want the AI to generate. 77 Token Limit.'),
52
  gr.Textbox(label='What you Do Not want the AI to generate.'),