#import
import gradio as gr
import torch

DESCRIPTION = '# Kandinsky 2.1'
if not torch.cuda.is_available():
    DESCRIPTION += '\n<p>Running on CPU 🥶 This demo does not work on CPU.</p>'

#description
description = """<div>
                CofAI.SD
              </div>
              """

#model
gr.Interface.load("models/stabilityai/stable-diffusion-2-1", description=description).launch()