Add app.py with Gradio interface for image generation

app.py

@@ -0,0 +1,32 @@
+import torch
+from diffusers import StableDiffusionPipeline, AutoencoderKL
+import gradio as gr
+
+repo = "IDKiro/sdxs-512-0.9"
+seed = 42
+weight_type = torch.float16
+
+# Load model.
+pipe = StableDiffusionPipeline.from_pretrained(repo, torch_dtype=weight_type)
+
+generator = pipe
+# move to GPU if available
+if torch.cuda.is_available():
+    generator = generator.to("cuda")
+
+
+def generate(prompts):
+    images = generator(list(prompts)).images
+    return [images]
+
+
+demo = gr.Interface(
+    generate,
+    "textbox",
+    "image",
+    batch=True,
+    max_batch_size=4,  # Set the batch size based on your CPU/GPU memory
+).queue()
+
+if __name__ == "__main__":
+    demo.launch()