Update app.py

app.py

@@ -1,3 +1,24 @@
+from diffusers import DiffusionPipeline
+import torch
 import gradio as gr
 
-gr.load("models/stabilityai/stable-diffusion-xl-base-1.0").launch()
+pipe = DiffusionPipeline.from_pretrained("stabilityai/stable-diffusion-xl-base-1.0", torch_dtype=torch.float16, use_safetensors=True, variant="fp16")
+pipe.to("cuda")
+
+def greet(prompt):
+    images = pipe(prompt=prompt).images[0]
+    print(images)
+    return images
+
+demo = gr.Interface(
+    fn=greet,
+    inputs=[
+        gr.Textbox(label="prompt")
+    ],
+    outputs=[gr.Image()],
+    api_name="chat",
+    clear_btn=None
+)
+
+if __name__ == "__main__":
+    demo.launch()