Update app_local.py

app_local.py

@@ -171,7 +171,7 @@ def polish_prompt(original_prompt: str) -> str:
             **model_inputs,
             max_new_tokens=256,
             do_sample=True,
-            temperature=0.5,
+            temperature=0.7,
             top_p=0.8,
             repetition_penalty=1.1,
             no_repeat_ngram_size=3,
@@ -245,9 +245,9 @@ def infer(
     prompt,
     seed=42,
     randomize_seed=False,
-    true_guidance_scale=1.0,
+    true_guidance_scale=4.0,
     num_inference_steps=8,
-    rewrite_prompt=False,
+    rewrite_prompt=True,
     num_images_per_prompt=1,
 ):
     """Image editing endpoint with optimized prompt handling"""
@@ -284,7 +284,7 @@ def infer(
             return pil_image  # Return original if resize fails
     
     # Add noise function for batch variation
-    def add_noise_to_image(pil_image, noise_level=0.02):
+    def add_noise_to_image(pil_image, noise_level=0.05):
         """Add slight noise to image to create variation in outputs"""
         try:
             if pil_image is None:
@@ -357,10 +357,10 @@ def infer(
                 generator = torch.Generator(device=device).manual_seed(base_seed + i*1000)
                 
                 # Add slight noise to the image for variation
-                noisy_image = add_noise_to_image(image, noise_level=0.01 + i*0.003)
+                noisy_image = add_noise_to_image(image, noise_level=0.05 + i*0.003)
                 
                 # Slightly vary guidance scale
-                varied_guidance = true_guidance_scale + random.uniform(-0.2, 0.2)
+                varied_guidance = true_guidance_scale + random.uniform(-0.5, 0.5)
                 varied_guidance = max(1.0, min(10.0, varied_guidance))
                 
                 # Generate single image with variations
@@ -460,7 +460,7 @@ with gr.Blocks(title="Qwen Image Edit - Fast Lightning Mode w/ Batch") as demo:
                         minimum=1.0, 
                         maximum=10.0, 
                         step=0.1, 
-                        value=3.0
+                        value=4.0
                     )
                     num_inference_steps = gr.Slider(
                         label="Inference Steps", 
@@ -487,7 +487,7 @@ with gr.Blocks(title="Qwen Image Edit - Fast Lightning Mode w/ Batch") as demo:
                 preview=True
             )
             prompt_info = gr.HTML(
-                value="<div style='padding:15px; background:#f8f9fa; border-radius:8px; margin-top:15px'>"
+                value="<div style='padding:15px; margin-top:15px'>"
                 "Prompt details will appear after generation</div>"
             )