Update app.py

app.py

@@ -29,16 +29,31 @@ def chat_fn(message, history):
     return tokenizer.decode(output[0], skip_special_tokens=True)
 
 # ---------------- IMAGE ----------------
-# ---------- IMAGE MODEL ----------
 from diffusers import StableDiffusionPipeline
+import torch
 
 img_pipe = StableDiffusionPipeline.from_pretrained(
     "stabilityai/sdxl-turbo",
-    torch_dtype=torch.float32
+    torch_dtype=torch.float32,
+    safety_checker=None
 )
 
+img_pipe.set_progress_bar_config(disable=True)
 img_pipe = img_pipe.to("cpu")
 
+def generate_image(prompt):
+    if not prompt.strip():
+        raise gr.Error("Prompt boş olamaz")
+
+    image = img_pipe(
+        prompt,
+        num_inference_steps=4,
+        guidance_scale=0.0
+    ).images[0]
+
+    return image
+
+
 
 # ---------------- UI ----------------
 with gr.Blocks(