Update app.py

app.py

@@ -206,7 +206,7 @@ css = """
 }
 """
 
-with gr.Blocks(css=css) as demo:
+with gr.Blocks() as demo:
     
     with gr.Column(elem_id="col-container"):
         gr.Markdown(f"""# FLUX.2 Klein [dev]
@@ -324,4 +324,4 @@ FLUX.2 Klein [dev] is a distilled model capable of generating, editing and combi
         outputs=[result, seed]
     )
 
-demo.launch()
+demo.launch(css=css)