app.py

# -*- coding: utf-8 -*-
"""StableDiffusion.ipynb

Automatically generated by Colaboratory.

Original file is located at
    https://colab.research.google.com/drive/1O7cgP1KO74ZqtTiW_ps2xZgEzwh-LuLi
"""

import gradio as gr
from diffusers import DiffusionPipeline
import torch

pipe = DiffusionPipeline.from_pretrained("stabilityai/stable-diffusion-xl-base-1.0", torch_dtype=torch.float32, use_safetensors=True, variant="fp16")
pipe.to('gpu')

def generate(prompt):
  image = pipe(prompt).images[0]
  return image

gr.Interface(fn=generate, inputs="text", outputs="image").launch(debug=True)