Upload 3 files

README.md

@@ -0,0 +1,40 @@
+---
+title: "AI Image Generator"
+emoji: 🖼️
+colorFrom: blue
+colorTo: purple
+sdk: gradio
+sdk_version: "6.4.0"
+python_version: "3.10"
+app_file: app.py
+pinned: false
+---
+
+# AI Image Generator (Multi-Models + Quality Settings)
+
+Простий застосунок для генерації зображень за допомогою різних моделей зі Stable Diffusion та Kandinsky.
+
+## 📌 Моделі
+- **Stable Diffusion v1.5** — класична модель, швидка і легка.
+- **Stable Diffusion XL Base** — сучасна, дає більш реалістичні результати.
+- **Kandinsky 2.2** — художній стиль, цікаві кольори та композиції.
+- **DreamShaper XL** — популярний fine-tune SDXL, більш деталізовані та креативні результати.
+
+## ⚙️ Налаштування
+- **Prompt** — текстовий опис бажаного зображення.
+- **Модель** — вибір генератора зі списку.
+- **Steps** — кількість кроків (більше = якісніше, але повільніше).
+- **Guidance scale** — наскільки точно модель слідує prompt-у (вище = точніше, нижче = креативніше).
+
+## 🚀 Використання
+1. Введіть опис (prompt).
+2. Оберіть модель генерації.
+3. Налаштуйте параметри якості.
+4. Натисніть кнопку **Submit** і отримайте результат.
+
+## 🛠️ Технології
+- [Gradio](https://gradio.app/) — інтерфейс.
+- [Diffusers](https://huggingface.co/docs/diffusers/index) — бібліотека для генерації.
+- [Stable Diffusion](https://huggingface.co/runwayml/stable-diffusion-v1-5) та інші моделі з Hugging Face Hub.
+
+---

app.py

@@ -0,0 +1,44 @@
+import gradio as gr
+from diffusers import StableDiffusionPipeline
+import torch
+
+# Список моделей для вибору
+IMAGE_MODELS = {
+    "Stable Diffusion v1.5": "runwayml/stable-diffusion-v1-5",
+    "Stable Diffusion XL Base": "stabilityai/stable-diffusion-xl-base-1.0",
+    "Kandinsky 2.2": "kandinsky-community/kandinsky-2-2-decoder",
+    "DreamShaper XL": "Lykon/DreamShaper"
+}
+
+# Кеш для завантажених пайплайнів
+loaded_pipes = {}
+
+def get_pipe(model_name):
+    if model_name not in loaded_pipes:
+        pipe = StableDiffusionPipeline.from_pretrained(
+            IMAGE_MODELS[model_name],
+            torch_dtype=torch.float16 if torch.cuda.is_available() else torch.float32
+        )
+        pipe = pipe.to("cuda" if torch.cuda.is_available() else "cpu")
+        loaded_pipes[model_name] = pipe
+    return loaded_pipes[model_name]
+
+def generate(prompt, model_name, steps, guidance):
+    pipe = get_pipe(model_name)
+    result = pipe(prompt, num_inference_steps=steps, guidance_scale=guidance)
+    return result.images[0]
+
+# Інтерфейс Gradio з вибором моделі та налаштуваннями якості
+demo = gr.Interface(
+    fn=generate,
+    inputs=[
+        gr.Textbox(label="Prompt", placeholder="Напиши опис сюди"),
+        gr.Dropdown(choices=list(IMAGE_MODELS.keys()), value="Stable Diffusion XL Base", label="Модель"),
+        gr.Slider(minimum=10, maximum=50, value=30, step=1, label="Кількість кроків (steps)"),
+        gr.Slider(minimum=1.0, maximum=15.0, value=7.5, step=0.5, label="Guidance scale")
+    ],
+    outputs=gr.Image(label="Result"),
+    title="AI Image Generator (Multi-Models + Quality Settings)"
+)
+
+demo.launch(ssr_mode=False)

requirements.txt

@@ -0,0 +1,5 @@
+diffusers==0.30.0
+transformers
+safetensors
+torch
+gradio==6.4.0