Update app.py

app.py

@@ -136,62 +136,61 @@ def tabs_select(e: gr.SelectData, _state):
 
 # Create Gradio interface
 with gr.Blocks(css=css) as demo:
-    gr.HTML("""\
-<p align="center"><img src="https://modelscope.oss-cn-beijing.aliyuncs.com/resource/qwen.png" style="height: 60px"/><p>"""
-            """<center><font size=8>📖 Qwen2-Math Demo</center>"""
-            """\
-<center><font size=3>This WebUI is based on Qwen2-VL for OCR and Qwen2-Math for mathematical reasoning. You can input either images or texts of mathematical or arithmetic problems.</center>"""
-            )
+    gr.HTML(
+        """<center><h1>Qwen2-Math Demo</h1><p>Use either uploaded images or sketches for math-related problems.</p></center>"""
+    )
     state = gr.State({"tab_index": 0})
     with gr.Row():
         with gr.Column():
             with gr.Tabs() as input_tabs:
                 with gr.Tab("Upload"):
-                    input_image = gr.Image(type="pil", label="Upload")
+                    input_image = gr.Image(type="pil", label="Upload Image")
                 with gr.Tab("Sketch"):
-                    input_sketchpad = gr.Sketchpad(type="pil", label="Sketch", layers=False)
-            input_tabs.select(fn=tabs_select, inputs=[state])
-            input_text = gr.Textbox(label="Input your question")
-            with gr.Row():
-                with gr.Column():
-                    clear_btn = gr.ClearButton([input_image, input_sketchpad, input_text])
-                with gr.Column():
-                    submit_btn = gr.Button("Submit", variant="primary")
+                    input_sketchpad = gr.Sketchpad(label="Sketch Pad")
+
+            input_tabs.select(fn=lambda e: {"tab_index": e.index}, inputs=[], outputs=state)
+            input_text = gr.Textbox(label="Your Question")
+            submit_btn = gr.Button("Submit")
+
         with gr.Column():
-            output_md = gr.Markdown(label="answer",
-                                    latex_delimiters=[{
-                                        "left": "\\(",
-                                        "right": "\\)",
-                                        "display": True
-                                    }, {
-                                        "left": "\\begin\{equation\}",
-                                        "right": "\\end\{equation\}",
-                                        "display": True
-                                    }, {
-                                        "left": "\\begin\{align\}",
-                                        "right": "\\end\{align\}",
-                                        "display": True
-                                    }, {
-                                        "left": "\\begin\{alignat\}",
-                                        "right": "\\end\{alignat\}",
-                                        "display": True
-                                    }, {
-                                        "left": "\\begin\{gather\}",
-                                        "right": "\\end\{gather\}",
-                                        "display": True
-                                    }, {
-                                        "left": "\\begin\{CD\}",
-                                        "right": "\\end\{CD\}",
-                                        "display": True
-                                    }, {
-                                        "left": "\\[",
-                                        "right": "\\]",
-                                        "display": True
-                                    }],
-                                    elem_id="qwen-md")
+            output_md = gr.Markdown(
+                label="Answer",
+                latex_delimiters=[{
+                    "left": "\\(",
+                    "right": "\\)",
+                    "display": True
+                }, {
+                    "left": "\\begin{equation}",
+                    "right": "\\end{equation}",
+                    "display": True
+                }, {
+                    "left": "\\begin{align}",
+                    "right": "\\end{align}",
+                    "display": True
+                }, {
+                    "left": "\\begin{alignat}",
+                    "right": "\\end{alignat}",
+                    "display": True
+                }, {
+                    "left": "\\begin{gather}",
+                    "right": "\\end{gather}",
+                    "display": True
+                }, {
+                    "left": "\\begin{CD}",
+                    "right": "\\end{CD}",
+                    "display": True
+                }, {
+                    "left": "\\[",
+                    "right": "\\]",
+                    "display": True
+                }],
+                elem_id="qwen-md"
+            )
+
         submit_btn.click(
             fn=math_chat_bot,
             inputs=[input_image, input_sketchpad, input_text, state],
-            outputs=output_md)
+            outputs=output_md,
+        )
 
 demo.launch()