Update app.py

app.py

@@ -13,27 +13,23 @@ chat_model = AutoModelForCausalLM.from_pretrained(
 )
 
 def chat_fn(message, history):
-    messages = [{"role": "user", "content": message}]
     prompt = tokenizer.apply_chat_template(
-        messages,
+        [{"role": "user", "content": message}],
         tokenize=False,
         add_generation_prompt=True
     )
-
     inputs = tokenizer(prompt, return_tensors="pt")
     output = chat_model.generate(
         **inputs,
         max_new_tokens=200,
+        do_sample=True,
         temperature=0.7
     )
     return tokenizer.decode(output[0], skip_special_tokens=True)
 
 # ---------------- IMAGE ----------------
-from diffusers import StableDiffusionPipeline
-import torch
-
 img_pipe = StableDiffusionPipeline.from_pretrained(
-    "stabilityai/sdxl-turbo",
+    "runwayml/stable-diffusion-v1-5",
     torch_dtype=torch.float32,
     safety_checker=None
 )
@@ -47,37 +43,29 @@ def generate_image(prompt):
 
     image = img_pipe(
         prompt,
-        num_inference_steps=4,
-        guidance_scale=0.0
+        num_inference_steps=20
     ).images[0]
 
     return image
 
-
-
 # ---------------- UI ----------------
-with gr.Blocks(
-    title="MaindAI",
-    theme=gr.themes.Soft(primary_hue="blue")
-) as app:
-
-    gr.Image(
-        value="Maindai.png",
-        show_label=False,
-        height=140
-    )
-
+with gr.Blocks(title="MaindAI") as app:
     gr.Markdown(
-        "<h1 style='text-align:center;color:#4da6ff;'>MaindAI</h1>"
+        """
+        <div style="text-align:center">
+            <img src="file=Maindai.png" width="140"><br>
+            <h1 style="color:#00bfff;">MaindAI</h1>
+        </div>
+        """
     )
 
     with gr.Row():
-        with gr.Column(scale=1):
+        with gr.Column():
             gr.ChatInterface(chat_fn)
-
-        with gr.Column(scale=1):
-            prompt = gr.Textbox(label="Görsel açıklaması")
-            btn = gr.Button("🎨 Oluştur")
+        with gr.Column():
+            prompt = gr.Textbox(label="🎨 Görsel Prompt")
+            btn = gr.Button("Oluştur")
             output = gr.Image()
             btn.click(generate_image, prompt, output)
 
+app.launch()