load_diffusers / app.py
shubhamofbce's picture
Upload folder using huggingface_hub
ced0dbb verified
import torch
import gradio as gr
from diffusers import StableDiffusionPipeline
pipe = StableDiffusionPipeline.from_pretrained(
"runwayml/stable-diffusion-v1-5",
torch_dtype=torch.float32,
use_safetensors=True,
)
demo = gr.Interface.from_pipeline(pipe)
demo.launch()