Update app.py

app.py

@@ -4,6 +4,8 @@ from transformers.image_utils import load_image
 from threading import Thread
 import torch
 import spaces
+from serve_constants import html_header, bibtext, learn_more_markdown, tos_markdown
+
 
 MODEL_ID = "TIGER-Lab/VL-Rethinker-7B"  
 processor = AutoProcessor.from_pretrained(MODEL_ID, trust_remote_code=True)
@@ -90,6 +92,8 @@ examples = [
     [{"text": "Solve this question.", "files": ["example_images/document.png"]}]
 ]
 
+gr.HTML(html_header)
+
 demo = gr.ChatInterface(
     fn=model_inference,
     description="# **VL-Rethinker-7B**",
@@ -101,4 +105,8 @@ demo = gr.ChatInterface(
     cache_examples=False,
 )
 
+gr.Markdown(tos_markdown)
+gr.Markdown(learn_more_markdown)
+gr.Markdown(bibtext)
+
 demo.launch(debug=True)