Update app.py

app.py

@@ -1,11 +1,19 @@
-import os
+import gradio as gr
+from diffusers import StableDiffusionPipeline
+import torch
 
-# Клонируем репозиторий
-os.system("git clone https://github.com/lllyasviel/stable-diffusion-webui-forge.git")
+# Используем официальную модель
+model_id = "CompVis/stable-diffusion-v1-4"
+pipe = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float16)
+pipe.to("cuda" if torch.cuda.is_available() else "cpu")
 
-# Перемещаемся в директорию с клонированным репозиторием
-os.chdir("stable-diffusion-webui-forge")
-
-# Выполняем другие необходимые команды, например, установка зависимостей
-os.system("pip install -r stable-diffusion-webui-forge/requirements.txt")
+def generate_image(prompt):
+    image = pipe(prompt).images[0]
+    return image
 
+gr.Interface(
+    fn=generate_image, 
+    inputs="text", 
+    outputs="image",
+    title="Stable Diffusion WebUI"
+).launch()