update app.py

app.py

@@ -50,20 +50,20 @@ def write2file(path, content):
 def sort_models(models):
     def custom_sort_key(model_name):
         # InternVL-Chat-V1-5 should be the first item
-        if model_name == "InternVL-Chat-V1-5":
-            return (1, model_name)  # 1 indicates highest precedence
-        elif model_name.startswith("InternVL-Chat-V1-5-"):
-            return (1, model_name)  # 1 indicates highest precedence
+        if model_name == "InternVL2-Pro":
+            return (2, model_name)  # 2 indicates highest precedence
+        elif model_name.startswith("InternVL2-8B"):
+            return (1, model_name)  # 0 indicates highest precedence
         else:
             return (0, model_name)  # 0 indicates normal order
 
     models.sort(key=custom_sort_key, reverse=True)
-    try:  # We have five InternVL-Chat-V1-5 models, randomly choose one to be the first
-        first_three = models[:4]
-        random.shuffle(first_three)
-        models[:4] = first_three
-    except:
-        pass
+    # try:  # We have five InternVL-Chat-V1-5 models, randomly choose one to be the first
+    #     first_three = models[:4]
+    #     random.shuffle(first_three)
+    #     models[:4] = first_three
+    # except:
+    #     pass
     return models
 
 
@@ -274,7 +274,7 @@ def change_system_prompt(state, system_prompt, request: gr.Request):
 def add_text(state, message, system_prompt, model_selector, request: gr.Request):
     print(f"state: {state}")
     if not state:
-        state, model_selector = load_demo_refresh_model_list(request)
+        state = init_state()
     images = message.get("files", [])
     text = message.get("text", "").strip()
     logger.info(f"add_text. ip: {request.client.host}. len: {len(text)}")
@@ -577,6 +577,7 @@ def build_demo(embed_mode):
         theme=gr.themes.Default(),
         css=block_css,
     ) as demo:
+        models = get_model_list()
         state = gr.State()
 
         if not embed_mode:
@@ -672,11 +673,9 @@ def build_demo(embed_mode):
                         [
                             {
                                 "files": [
-                                    "gallery/child_1.jpg",
-                                    "gallery/child_2.jpg",
-                                    f"gallery/child_3.jpg",
+                                    "gallery/water.jpg",
                                 ],
-                                "text": "这三帧图片讲述了一件什么事情？",
+                                "text": "Please describe this image.",
                             }
                         ],
                     ],
@@ -847,8 +846,6 @@ if __name__ == "__main__":
     if not args.worker_ip:
         args.worker_ip = os.environ.get("WORKER_IP", None)
 
-    models = get_model_list()
-
     sd_worker_url = args.sd_worker_url
     logger.info(args)
     demo = build_demo(args.embed)