Update app.py

app.py

@@ -88,25 +88,26 @@ def model_inference(input_dict, history):
         buffer += new_text
         yield buffer
 
-examples = [
-    [{"text": "Solve this question.", "files": ["example_images/document.png"]}]
-]
-
-gr.HTML(html_header)
-
-demo = gr.ChatInterface(
-    fn=model_inference,
-    description="# **VL-Rethinker-7B**",
-    examples=examples,
-    fill_height=True,
-    textbox=gr.MultimodalTextbox(label="Query Input", file_types=["image"], file_count="multiple"),
-    stop_btn="Stop Generation",
-    multimodal=True,
-    cache_examples=False,
-)
+with gr.Blocks() as demo:
+    examples = [
+        [{"text": "Solve this question.", "files": ["example_images/document.png"]}]
+    ]
 
-gr.Markdown(tos_markdown)
-gr.Markdown(learn_more_markdown)
-gr.Markdown(bibtext)
+    gr.HTML(html_header)
+    
+    gr.ChatInterface(
+        fn=model_inference,
+        description="# **VL-Rethinker-7B**",
+        examples=examples,
+        fill_height=True,
+        textbox=gr.MultimodalTextbox(label="Query Input", file_types=["image"], file_count="multiple"),
+        stop_btn="Stop Generation",
+        multimodal=True,
+        cache_examples=False,
+    )
+    
+    gr.Markdown(tos_markdown)
+    gr.Markdown(learn_more_markdown)
+    gr.Markdown(bibtext)
 
 demo.launch(debug=True)