Update app.py

app.py

@@ -12,7 +12,7 @@ import base64
 import json
 
 from huggingface_hub import login
-from prompt_augment import PromptAugment
+# from prompt_augment import PromptAugment
 login(token=os.environ.get('hf'))
 
 
@@ -24,23 +24,23 @@ device = "cuda" if torch.cuda.is_available() else "cpu"
 
 # Load the model pipeline
 pipe = QwenImageEditPlusPipeline.from_pretrained("FireRedTeam/FireRed-Image-Edit-1.0", torch_dtype=dtype).to(device)
-prompt_handler = PromptAugment()
+# prompt_handler = PromptAugment()
 
 # --- UI Constants and Helpers ---
 MAX_SEED = np.iinfo(np.int32).max
 
 # --- Main Inference Function (with hardcoded negative prompt) ---
-@spaces.GPU(duration=180)
+@spaces.GPU()
 def infer(
     images,
     prompt,
-    seed=42,
-    randomize_seed=False,
+    seed=5555,
+    randomize_seed=True,
     true_guidance_scale=1.0,
     num_inference_steps=50,
     height=None,
     width=None,
-    rewrite_prompt=True,
+    rewrite_prompt=False,
     num_images_per_prompt=1,
     progress=gr.Progress(track_tqdm=True),
 ):
@@ -75,9 +75,9 @@ def infer(
     print(f"Calling pipeline with prompt: '{prompt}'")
     print(f"Negative Prompt: '{negative_prompt}'")
     print(f"Seed: {seed}, Steps: {num_inference_steps}, Guidance: {true_guidance_scale}, Size: {width}x{height}")
-    if rewrite_prompt and len(pil_images) > 0:
+    if False and rewrite_prompt and len(pil_images) > 0:
         # prompt = polish_prompt(prompt, pil_images[0])
-        prompt = prompt_handler.predict(prompt, [pil_images[0]])
+        # prompt = prompt_handler.predict(prompt, [pil_images[0]])
         print(f"Rewritten Prompt: {prompt}")
     
 
@@ -96,25 +96,22 @@ def infer(
 
     return image, seed
 
-# --- Examples and UI Layout ---
-examples = []
-
 css = """
-#col-container {
+#NOcol-container {
     margin: 0 auto;
     max-width: 1024px;
 }
-#edit_text{margin-top: -62px !important}
 """
 
 def get_image_base64(image_path):
     with open(image_path, "rb") as img_file:
         return base64.b64encode(img_file.read()).decode('utf-8')
+
 logo_base64 = get_image_base64("logo.png")
 
 with gr.Blocks(css=css) as demo:
     with gr.Column(elem_id="col-container"):
-        gr.HTML(f'<img src="data:image/png;base64,{logo_base64}" alt="Firered Logo" width="400" style="display: block; margin: 0 auto;">')
+        gr.HTML(f'<img src="data:image/png;base64,{logo_base64}" alt="FireRedTeam Logo" width="400" />')
         gr.Markdown("[Learn more](https://github.com/FireRedTeam/FireRed-Image-Edit) about the FireRed-Image-Edit series.")
         with gr.Row():
             with gr.Column():
@@ -129,7 +126,7 @@ with gr.Blocks(css=css) as demo:
                     placeholder="describe the edit instruction",
                     container=False,
             )
-            run_button = gr.Button("Edit!", variant="primary")
+            run_button = gr.Button("Edit", variant="primary")
 
         with gr.Accordion("Advanced Settings", open=False):
             # Negative prompt UI element is removed here
@@ -167,7 +164,7 @@ with gr.Blocks(css=css) as demo:
                     minimum=256,
                     maximum=2048,
                     step=8,
-                    value=None,
+                    value=1024,
                 )
                 
                 width = gr.Slider(
@@ -175,7 +172,7 @@ with gr.Blocks(css=css) as demo:
                     minimum=256,
                     maximum=2048,
                     step=8,
-                    value=None,
+                    value=1024,
                 )
                 
                 
@@ -202,4 +199,4 @@ with gr.Blocks(css=css) as demo:
 
 if __name__ == "__main__":
     # demo.launch()
-    demo.launch(allowed_paths=["./"]) 
+    demo.launch(concurrency_limit=2)