Update app.py

app.py

@@ -17,7 +17,7 @@ from optimization import optimize_pipeline_
 
 MAX_SEED = np.iinfo(np.int32).max
 
-pipe = FluxKontextPipeline.from_pretrained("black-forest-labs/FLUX.1-Kontext-dev", torch_dtype=torch.bfloat16).to("cuda")
+pipe = FluxKontextPipeline.from_pretrained("black-forest-labs/FLUX.1-Kontext-dev", torch_dtype=torch.bfloat16).to("cpu")
 optimize_pipeline_(pipe, image=Image.new("RGB", (512, 512)), prompt='prompt')
 
 @spaces.GPU
@@ -89,18 +89,12 @@ def infer_example(input_image, prompt):
     image, seed, _ = infer(input_image, prompt)
     return image, seed
 
-css="""
-#col-container {
-    margin: 0 auto;
-    max-width: 960px;
-}
-"""
 
-with gr.Blocks(css=css) as demo:
+with gr.Blocks(theme="Yntec/HaleyCH_Theme_Orange") as demo:
     
     with gr.Column(elem_id="col-container"):
-        gr.Markdown(f"""# FLUX.1 Kontext [dev]
-Image editing and manipulation model guidance-distilled from FLUX.1 Kontext [pro], [[blog]](https://bfl.ai/announcements/flux-1-kontext-dev) [[model]](https://huggingface.co/black-forest-labs/FLUX.1-Kontext-dev)
+        gr.Markdown(f"""# Image Editor
+        ⚠ Sorry for the inconvenience. The Space is currently running on the CPU, which might affect performance. We appreciate your understanding.
         """)
         with gr.Row():
             with gr.Column():
@@ -145,19 +139,6 @@ Image editing and manipulation model guidance-distilled from FLUX.1 Kontext [pro
             with gr.Column():
                 result = gr.Image(label="Result", show_label=False, interactive=False)
                 reuse_button = gr.Button("Reuse this image", visible=False)
-        
-            
-        examples = gr.Examples(
-            examples=[
-                ["flowers.png", "turn the flowers into sunflowers"],
-                ["monster.png", "make this monster ride a skateboard on the beach"],
-                ["cat.png", "make this cat happy"]
-            ],
-            inputs=[input_image, prompt],
-            outputs=[result, seed],
-            fn=infer_example,
-            cache_examples="lazy"
-        )
             
     gr.on(
         triggers=[run_button.click, prompt.submit],