from diffusers import StableDiffusionPipeline, StableDiffusionUpscalePipeline
import gradio as gr

# pipe = StableDiffusionPipeline.from_pretrained("CompVis/stable-diffusion-v1-4")
pipe = StableDiffusionUpscalePipeline.from_pretrained("stabilityai/stable-diffusion-x4-upscaler")


gr.Interface.from_pipeline(pipe).launch()