Update app.py

app.py

@@ -71,7 +71,7 @@ gvlabtheme = OpenGVLab(primary_hue=colors.blue,
 def main():
     parser = argparse.ArgumentParser()
     parser.add_argument('--model_type', default="llama", type=str)
-    parser.add_argument('--base_model', default=r"/data/wangpeng/JiaotongGPT-main/merged-sft-no-1ep", type=str)
+    parser.add_argument('--base_model', default="DUOMO-Lab/TransGPT-v0", type=str)
     parser.add_argument('--lora_model', default="", type=str, help="If None, perform inference on the base model")
     parser.add_argument('--tokenizer_path', default=None, type=str)
     parser.add_argument('--gpus', default="0", type=str)
@@ -208,10 +208,6 @@ def main():
     with gr.Blocks(title="DUOMO TransGPT!", theme=gvlabtheme,
                    css="#chatbot {overflow:auto; height:500px;} #InputVideo {overflow:visible; height:320px;} footer {visibility: none}") as demo:
         gr.Markdown(title)
-    # with gr.Blocks() as demo:
-    #     gr.HTML("""<h1 align="center">TransGPT</h1>""")
-    #     # gr.Markdown(
-    #     #     "> 为了促进医疗行业大模型的开放研究，本项目开源了TransGPT医疗大模型")
         chatbot = gr.Chatbot()
         with gr.Row():
             with gr.Column(scale=4):