| #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() |