Create app.py

app.py

@@ -0,0 +1,17 @@
+import torch 
+import gradio as gr
+from diffusers import StableDiffusionPipeline 
+
+class CFG:
+    device = 'cpu'
+    seed = 42
+    Generator = torch.Generator(device).manual_seed(seed)
+    image_gen_steps = 35 
+    image_gen_model_id = 'stabilityai/stable-diffusion-2'
+    image_gen_size = (400, 400)
+    image_gen_guidance_scale = 9
+
+image_gen_model = StableDiffusionPipeline.from_pretrained(
+    CFG.image_gen_model_id, torch_dtype=torch.float32,
+    revision="fp16", use_auth_token='hf_'
+)