Update app.py

app.py

@@ -19,12 +19,30 @@ def infer(prompt):
     image = pipe(prompt, guidance_scale=7.5).images[0].resize((512,512))
     return image
 
-gr.Interface(
-    fn = infer,
-    inputs = [
-        gr.Textbox(value="Two cats playing chess on a tree branch")
-    ],
-    outputs = [
-        gr.Image()
-    ]
-).launch()
+css = """
+#col-container{
+    max-width: 720px;
+}
+"""
+with gr.Blocks(css=css) as demo:
+    with gr.Column(elem_id="col-container"):
+        gr.HTML("""
+        <h2 style="text-align: center;">
+            SDXL Using Direct Preference Optimization
+        </h2>
+        """)
+        prompt_in = gr.Textbox(label="Prompt", value="An old man with a bird on his head)
+        submit_btn = gr.Button("Submit")
+        result = gr.Image(label="DPO SDXL Result")
+
+    submit_btn.click(
+        fn = infer,
+        inputs = [
+            prompt_in
+        ],
+        outputs = [
+            result
+        ]
+    )
+
+demo.queue().launch()