Spaces:
Runtime error
Runtime error
Update app.py
Browse files
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.'),
|