Update app.py

app.py

@@ -10,13 +10,10 @@ from libra.eval.run_libra import load_model
 DEFAULT_MODEL_PATH = "X-iZhang/libra-v1.0-7b"
 
 def get_model_short_name(model_path: str) -> str:
-    """
-    提取模型路径最后一个 '/' 之后的部分，作为在下拉菜单中显示的名字。
-    例如: "X-iZhang/libra-v1.0-7b" -> "libra-v1.0-7b"
-    """
+
     return model_path.rstrip("/").split("/")[-1]
 
-# 全局/或在main里定义都行，这里示例放在外层
+
 loaded_models = {}  # {model_key: reuse_model_object}
 
 def generate_radiology_description(
@@ -31,32 +28,27 @@ def generate_radiology_description(
     max_new_tokens: int,
     model_paths_dict: dict
 ) -> str:
-    """
-    执行放射学报告推理:
-    1) 根据下拉选的模型名称 -> 找到实际 model_path
-    2) 确保用户选了 Current & Prior 图片
-    3) 调用 libra_eval
-    """
+
     real_model_path = model_paths_dict[selected_model_name]
 
-    # 若用户没选/没上传 Current Image，一定报错
+
     if not current_img_data:
         return "Error: Please select or upload the Current Image."
 
-    # 如果用户勾选了 without prior image，就把 prior_img_data 设为 current_img_data
+
     if use_no_prior:
         prior_img_data = current_img_data
     else:
-        # 未勾选时，需要prior_img_data
+
         if not prior_img_data:
             return "Error: Please select or upload the Prior Image, or check 'Without Prior Image'."
 
-    # 若已经加载过该模型，则直接复用
+
     if selected_model_name in loaded_models:
         reuse_model = loaded_models[selected_model_name]
     else:
         reuse_model = load_model(real_model_path)
-        # 缓存起来
+
         loaded_models[selected_model_name] = reuse_model
         
     try:
@@ -77,23 +69,21 @@ def generate_radiology_description(
         return f"An error occurred during model inference: {str(e)}"
 
 def main():
-    # ========== 获取当前脚本 (app.py) 所在目录 ==========
+
     cur_dir = os.path.abspath(os.path.dirname(__file__))
 
-    # ========== 准备本地示例图片的绝对路径 ==========
-    # 向上回退两级： app.py -> serve/ -> libra/ -> Libra/ (同级)
     example_curent_path = os.path.join(cur_dir, "examples", "curent.jpg")
     example_curent_path = os.path.abspath(example_curent_path)
 
     example_prior_path = os.path.join(cur_dir, "examples", "prior.jpg")
     example_prior_path = os.path.abspath(example_prior_path)
 
-    # Gradio Examples 要求：对单个 gr.Image 而言，每个示例写成 ["本地文件路径"]
+
     IMAGE_EXAMPLES = [
         [example_curent_path],
         [example_prior_path]
     ]
-    # ========== 命令行解析 (可选) ==========
+   
     parser = argparse.ArgumentParser(description="Demo for Radiology Image Description Generator (Local Examples)")
     parser.add_argument(
         "--model-path",
@@ -104,21 +94,19 @@ def main():
     args = parser.parse_args()
     cmd_model_path = args.model_path
     
-    # ========== 设置多模型下拉菜单 ==========
+    
     model_paths_dict = {}
     user_key = get_model_short_name(cmd_model_path)
     model_paths_dict[user_key] = cmd_model_path
 
-    # 如果用户传入的模型 != 默认模型，则加上默认模型选项
+   
     if cmd_model_path != DEFAULT_MODEL_PATH:
         default_key = get_model_short_name(DEFAULT_MODEL_PATH)
         model_paths_dict[default_key] = DEFAULT_MODEL_PATH
 
-    # （可选）若想预先加载模型，避免重复加载，可在此处:
-    # reuse_model = load_model(cmd_model_path)
-    # 然后在 generate_radiology_description 里改造传 reuse_model
+
     
-    # ========== 搭建 Gradio 界面 ==========
+
     with gr.Blocks(title="Libra: Radiology Analysis with Direct URL Examples") as demo:
         gr.Markdown("""
         ## 🩻 Libra: Leveraging Temporal Images for Biomedical Radiology Analysis
@@ -127,7 +115,7 @@ def main():
         **Requires a GPU to run effectively!**
         """)
 
-        # 下拉模型选择
+        
         model_dropdown = gr.Dropdown(
             label="Select Model",
             choices=list(model_paths_dict.keys()),
@@ -135,7 +123,7 @@ def main():
             interactive=True
         )
         
-        # 临床Prompt
+        
         prompt_input = gr.Textbox(
             label="Clinical Prompt",
             value="Provide a detailed description of the findings in the radiology image.",
@@ -147,7 +135,7 @@ def main():
             )
         )
 
-        # Current & Prior 画像
+        
         with gr.Row():
             with gr.Column():
                 gr.Markdown("### Current Image")
@@ -235,7 +223,7 @@ def main():
             outputs=output_text
         )
         
-        # 界面底部插入条款说明
+        
         gr.Markdown("""
         ### Terms of Use