Update app.py

app.py

@@ -52,7 +52,7 @@ pipe.load_lora_weights(
     )
 pipe.fuse_lora()
 
-pipe.transformer.__class__ = QwenImageTransformer2DModel
+# pipe.transformer.__class__ = QwenImageTransformer2DModel
 pipe.transformer.set_attn_processor(QwenDoubleStreamAttnProcessorFA3())
 
 
@@ -90,13 +90,21 @@ examples = [
     "replace the object with a flower vase",
 ]
 
-css="""
+css = """
 #col-container {
     margin: 0 auto;
-    max-width: 1000px;
+    max-width: 1024px;
 }
+#logo-title {
+    text-align: center;
+}
+#logo-title img {
+    width: 400px;
+}
+#edit_text{margin-top: -62px !important}
 """
 
+
 with gr.Blocks(css=css) as demo:
     
     with gr.Column(elem_id="col-container"):
@@ -137,7 +145,8 @@ with gr.Blocks(css=css) as demo:
                     max_lines=1,
                     placeholder="Enter what you don't want (optional)",
                     container=False,
-                    value=""
+                    value="",
+                    visible=False
                 )
                 run_button = gr.Button("Run")
                 
@@ -176,7 +185,7 @@ with gr.Blocks(css=css) as demo:
   
                 num_inference_steps = gr.Slider(
                     label="Number of inference steps",
-                    minimum=10,
+                    minimum=1,
                     maximum=50,
                     step=1,
                     value=8,