Update app.py

app.py

@@ -886,7 +886,7 @@ css = """
 }
 """
 
-with gr.Blocks(css=css, theme=gr.themes.Citrus()) as demo:
+with gr.Blocks() as demo:
     
     with gr.Column(elem_id="col-container"):
         gr.Markdown("""# FLUX.2 [Klein] Relight Brush 🖌️
@@ -1050,4 +1050,4 @@ Upload an image → draw colored strokes where you want to add/change lighting
         outputs=[result, seed]
     )
 
-demo.launch()
+demo.launch(css=css, theme=gr.themes.Citrus())