Update app.py

app.py

@@ -170,7 +170,6 @@ def infer(
     aspect_ratio="1:1",
     guidance_scale=1.0,
     num_inference_steps=8,
-    prompt_enhance=True,
     progress=gr.Progress(track_tqdm=True),
 ):
     """
@@ -185,7 +184,7 @@ def infer(
             encourages the model to generate images that are more closely related 
             to the prompt.
         num_inference_steps (int): The number of denoising steps.
-        prompt_enhance (bool): If True, the prompt is rewritten by an external 
+    
             LLM to add more detail.
         progress (gr.Progress): A Gradio Progress object to track the generation
             progress in the UI.
@@ -207,8 +206,7 @@ def infer(
     generator = torch.Generator(device="cuda").manual_seed(seed)
     
     print(f"Calling pipeline with prompt: '{prompt}'")
-    if prompt_enhance:
-        prompt = rewrite(prompt)
+
         
     print(f"Actual Prompt: '{prompt}'")
     print(f"Negative Prompt: '{negative_prompt}'")
@@ -290,7 +288,6 @@ with gr.Blocks(css=css) as demo:
                     choices=["1:1", "16:9", "9:16", "4:3", "3:4", "3:2", "2:3"],
                     value="16:9",
                 )
-                prompt_enhance = gr.Checkbox(label="Prompt Enhance", value=True)
 
             with gr.Row():
                 guidance_scale = gr.Slider(
@@ -321,7 +318,6 @@ with gr.Blocks(css=css) as demo:
             aspect_ratio,
             guidance_scale,
             num_inference_steps,
-            prompt_enhance,
         ],
         outputs=[result, seed],
     )