Update app.py

app.py

@@ -45,15 +45,14 @@ def generate_image(prompt: str, seed: int = 42, aspect_ratio: str = "16:9", rand
 
     width, height = ASPECT_RATIOS.get(aspect_ratio, (DEFAULT_WIDTH, DEFAULT_HEIGHT))
 
-    # Allow prompts in Albanian or English and still enhance them
-    enhanced_prompt = prompt.strip() + ", ultra realistic, sharp, 8k resolution, cinematic lighting"
+    prompt += ", ultra realistic, sharp, 8k resolution, cinematic lighting"
 
     try:
         generator = torch.Generator(device=device).manual_seed(int(float(seed)))
         start_time = time.time()
         with torch.inference_mode():
             image = pipe(
-                prompt=enhanced_prompt,
+                prompt=prompt,
                 width=width,
                 height=height,
                 num_inference_steps=INFERENCE_STEPS,
@@ -75,23 +74,22 @@ examples = [
     "Pamje nga një shtëpi moderne në stilin Minecraft"
 ]
 
-# --- App Layout ---
 with gr.Blocks(css="""
 body::before {
     content: "";
     display: block;
     height: 640px;
-    background-color: #0f1117;
+    background-color: inherit;
 }
 button[aria-label="Download"] {
-    transform: scale(1.5);
+    transform: scale(3);
     transform-origin: top right;
     margin: 0 !important;
     padding: 6px !important;
 }
 """) as app:
     gr.Markdown("# 🖼️ Gjenerues Imazhesh FLUX")
-    gr.Markdown("Përdor modelin FLUX për të krijuar imazhe fantastike nga përshkrime në **gjuhën shqipe ose angleze**.")
+    gr.Markdown("Përdor modelin FLUX për të krijuar imazhe fantastike.")
 
     with gr.Row():
         with gr.Column(scale=2):