Kev09 commited on
Commit
ba79bbb
·
verified ·
1 Parent(s): a4b7992

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -1,5 +1,5 @@
1
  import gradio as gr
2
-
3
  from diffusers import DiffusionPipeline
4
 
5
  pipe = DiffusionPipeline.from_pretrained("CompVis/ldm-super-resolution-4x-openimages")
@@ -13,5 +13,5 @@ def greet(name):
13
  print("IMAGE: \n",image)
14
  return image
15
 
16
- demo = gr.Interface(fn=greet, inputs="text", outputs="image")
17
  demo.launch()
 
1
  import gradio as gr
2
+ from PIL import Image
3
  from diffusers import DiffusionPipeline
4
 
5
  pipe = DiffusionPipeline.from_pretrained("CompVis/ldm-super-resolution-4x-openimages")
 
13
  print("IMAGE: \n",image)
14
  return image
15
 
16
+ demo = gr.Interface(fn=greet, inputs="text", outputs=gr.Image(type="pil"))
17
  demo.launch()