Update app.py

app.py

@@ -1,16 +1,21 @@
 import gradio as gr
 import torch
 from torch import autocast
+from PIL import Image
 from diffusers import StableDiffusionPipeline
 
 print("hello sylvain")
 
 YOUR_TOKEN="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"
-
+device="cpu"
 pipe = StableDiffusionPipeline.from_pretrained("CompVis/stable-diffusion-v1-4", use_auth_token=YOUR_TOKEN)
-pipe.to("cpu")
+pipe.to(device)
+
+def infer(prompt):
+    image = pipe(prompt)["sample"][0]
+    return image
 
 def process(name):
   return "hello " + name
   
-gr.Interface(fn=process, inputs="text", outputs="text").launch()
+gr.Interface(fn=infer, inputs="text", outputs="image").launch()