Update app.py

app.py

@@ -24,7 +24,7 @@ pipe = Flux2Pipeline.from_pretrained(
 )
 
 # Enable memory optimizations for ZeroGPU
-pipe.enable_model_cpu_offload()  # This offloads models to CPU when not in use, saving VRAM
+pipe.enable_model_cpu_offload()
 
 print("Pipeline loaded successfully!")
 
@@ -53,7 +53,7 @@ def update_dimensions_from_image(image_list):
     
     return new_width, new_height
 
-@spaces.GPU(duration=60)  # Increased duration since we're doing text encoding + generation
+@spaces.GPU(duration=60)
 def infer(prompt, input_images=None, seed=42, randomize_seed=False, width=1024, height=1024, num_inference_steps=50, guidance_scale=2.5, progress=gr.Progress(track_tqdm=True)):
     
     if randomize_seed:
@@ -99,20 +99,20 @@ examples_images = [
     ["The person from image 1 is petting the cat from image 2, the bird from image 3 is next to them", ["woman1.webp", "cat_window.webp", "bird.webp"]]
 ]
 
-css="""
+css = """
 #col-container {
     margin: 0 auto;
     max-width: 1200px;
 }
-.gallery-container img{
+.gallery-container img {
     object-fit: contain;
 }
 """
 
-with gr.Blocks(css=css) as demo:
+with gr.Blocks() as demo:
     
     with gr.Column(elem_id="col-container"):
-        gr.Markdown(f"""# FLUX.2 [dev]
+        gr.Markdown("""# FLUX.2 [dev]
 FLUX.2 [dev] is a 32B model rectified flow capable of generating, editing and combining images based on text instructions model [[model](https://huggingface.co/black-forest-labs/FLUX.2-dev)], [[blog](https://bfl.ai/blog/flux-2)]
         """)
         with gr.Row():
@@ -214,4 +214,4 @@ FLUX.2 [dev] is a 32B model rectified flow capable of generating, editing and co
         outputs=[result, seed]
     )
 
-demo.launch()
+demo.launch(css=css)