Upload app.py

app.py

@@ -0,0 +1,40 @@
+import gradio as gr
+from diffusers import StableDiffusionPipeline
+import torch
+
+# Список доступних моделей
+MODELS = {
+    "Stable Diffusion v1.5": "runwayml/stable-diffusion-v1-5",
+    "Stable Diffusion 2.1 Base": "stabilityai/stable-diffusion-2-1-base",
+    "Stable Diffusion XL": "stabilityai/stable-diffusion-xl-base-1.0"
+}
+
+# Кеш для завантажених пайплайнів
+loaded_pipes = {}
+
+def get_pipe(model_name):
+    if model_name not in loaded_pipes:
+        pipe = StableDiffusionPipeline.from_pretrained(
+            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):
+    pipe = get_pipe(model_name)
+    result = pipe(prompt)
+    return result.images[0]
+
+demo = gr.Interface(
+    fn=generate,
+    inputs=[
+        gr.Textbox(label="Prompt", placeholder="Напиши опис сюди"),
+        gr.Dropdown(choices=list(MODELS.keys()), value="Stable Diffusion v1.5", label="Модель")
+    ],
+    outputs=gr.Image(label="Result"),
+    title="AI Image Generator (Multiple Models)"
+)
+
+demo.launch(ssr_mode=False)