Update app.py

app.py

@@ -24,7 +24,6 @@ pipe.transformer.load_state_dict(state_dict)
 @spaces.GPU(duration=120)
 def generate_image(
     prompt,
-    negative_prompt="",
     width=1024,
     height=1024,
     guidance_scale=3.5,
@@ -38,7 +37,6 @@ def generate_image(
     
     image = pipe(
         prompt=prompt,
-        negative_prompt=negative_prompt if negative_prompt else None,
         guidance_scale=guidance_scale,
         height=height,
         width=width,
@@ -49,9 +47,9 @@ def generate_image(
     
     return image, seed
 
-with gr.Blocks(title="FLUX SRPO Text-to-Image") as demo:
-    gr.Markdown("# FLUX with SRPO (Self-Regulating Preference Optimization)")
-    gr.Markdown("Generate high-quality images using FLUX model enhanced with Tencent's SRPO technique")
+with gr.Blocks(title="FLUX SRPO Text-to-Image", theme=gr.themes.Soft(primary_hue="blue", secondary_hue="gray", neutral_hue="slate")) as demo:
+    gr.Markdown("# Flux SRPO")
+    gr.Markdown("Generate images using FLUX model enhanced with Tencent's SRPO technique")
     
     with gr.Row():
         with gr.Column(scale=3):
@@ -60,11 +58,6 @@ with gr.Blocks(title="FLUX SRPO Text-to-Image") as demo:
                 placeholder="Describe the image you want to generate...",
                 lines=3
             )
-            negative_prompt = gr.Textbox(
-                label="Negative Prompt (optional)",
-                placeholder="What you don't want to see in the image...",
-                lines=2
-            )
             
             with gr.Row():
                 width = gr.Slider(
@@ -124,7 +117,7 @@ with gr.Blocks(title="FLUX SRPO Text-to-Image") as demo:
     
     generate_btn.click(
         fn=generate_image,
-        inputs=[prompt, negative_prompt, width, height, guidance_scale, num_inference_steps, seed],
+        inputs=[prompt, width, height, guidance_scale, num_inference_steps, seed],
         outputs=[output_image, used_seed]
     )