xhxhkxh commited on
Commit
7599fda
1 Parent(s): 811f076

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -1
app.py CHANGED
@@ -1,4 +1,9 @@
1
  from diffusers import DiffusionPipeline
2
  import gradio as gr
 
 
 
3
 
4
- pipeline = DiffusionPipeline.from_pretrained("PublicPrompts/All-In-One-Pixel-Model")
 
 
 
1
  from diffusers import DiffusionPipeline
2
  import gradio as gr
3
+ def gen(text):
4
+ pipeline = DiffusionPipeline.from_pretrained("PublicPrompts/All-In-One-Pixel-Model")
5
+ return pipeline(text).images[0]
6
 
7
+ gr.Interface(fn=gen,inputs='text'
8
+ outputs='image'
9
+ )