Create app.py

app.py

@@ -1,27 +1,14 @@
-# app.py файлын жасаңыз
 import gradio as gr
 from diffusers import StableDiffusionPipeline
 import torch
 
 model_id = "runwayml/stable-diffusion-v1-5"
-pipe = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float16)
-pipe = pipe.to("cuda")
+pipe = StableDiffusionPipeline.from_pretrained(model_id)
+pipe = pipe.to("cuda" if torch.cuda.is_available() else "cpu")
 
-def generate_image(prompt):
-    image = pipe(
-        prompt,
-        num_inference_steps=30,
-        guidance_scale=7.5
-    ).images[0]
+def generate(prompt):
+    image = pipe(prompt).images[0]
     return image
 
-# Gradio интерфейсі
-iface = gr.Interface(
-    fn=generate_image,
-    inputs=gr.Textbox(label="Сурет сипаттамасы", lines=2),
-    outputs=gr.Image(label="Нәтиже"),
-    title="Stable Diffusion генераторы",
-    description="Мәтіндік сипаттама бойынша кескін жасаңыз"
-)
-
+iface = gr.Interface(fn=generate, inputs="text", outputs="image")
 iface.launch()