Update app.py
Browse files
app.py
CHANGED
@@ -1,4 +1,9 @@
|
|
1 |
from diffusers import DiffusionPipeline
|
2 |
import gradio as gr
|
|
|
|
|
|
|
3 |
|
4 |
-
|
|
|
|
|
|
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 |
+
)
|