Update app.py

app.py

@@ -143,6 +143,8 @@ class InferenceDemo(object):
             conv_mode = "qwen_1_5"
         elif "pangea" in model_name.lower():
             conv_mode = "qwen_1_5"
+        elif "mammoth-vl" in model_name.lower():
+            conv_mode = "qwen_2_5"
         else:
             conv_mode = "llava_v0"
 
@@ -411,7 +413,7 @@ def bot(history, temperature, top_p, max_output_tokens):
     with open(get_conv_log_filename(), "a") as fout:
         data = {
             "type": "chat",
-            "model": "Pangea-7b",
+            "model": "MAmmoTH-VL-8b",
             "state": history,
             "images": all_image_hash,
             "images_path": all_image_path
@@ -478,7 +480,7 @@ with gr.Blocks(
                     label="Max output tokens",
                 )
         with gr.Row():
-            chatbot = gr.Chatbot([], elem_id="Pangea", bubble_full_width=False, height=750)
+            chatbot = gr.Chatbot([], elem_id="MAmmoTH-VL-8B", bubble_full_width=False, height=750)
 
         with gr.Row():
             upvote_btn = gr.Button(value="👍  Upvote", interactive=True)
@@ -699,7 +701,7 @@ if __name__ == "__main__":
     argparser.add_argument("--server_name", default="0.0.0.0", type=str)
     argparser.add_argument("--port", default="6123", type=str)
     argparser.add_argument(
-        "--model_path", default="neulab/Pangea-7B", type=str
+        "--model_path", default="MMSFT/MAmmoTH-VL-8B", type=str
     )
     # argparser.add_argument("--model-path", type=str, default="facebook/opt-350m")
     argparser.add_argument("--model-base", type=str, default=None)