Update app.py

app.py

@@ -184,7 +184,7 @@ if __name__ == "__main__":
     
     # YoloV8、V9官方模型列表，从左往右由小到大，第一次使用会自动下载
     model_list = ["yolov8n.pt", "yolov9c.pt", "yolov9e", "yolov8s.pt", "yolov8m.pt", "yolov8l.pt", "yolov8x.pt"]
-    
+    model_list1 = ["FATransformer","UFGS","ALAT"]
     # 获取YoloV8模型可以检测的所有类别，默认调用model_list中第一个模型
     detect_classes = get_detectable_classes(model_list[0])
     
@@ -207,7 +207,7 @@ if __name__ == "__main__":
                 # 列容器，垂直排列元素
                 with gr.Column():
                     input_path = gr.Video(label="Input video") # 视频输入控件，用于上传视频文件
-                    model = gr.Dropdown(model_list, value=0, label="Model") # 下拉菜单控件，用于选择模型
+                    model = gr.Dropdown(model_list1, value=0, label="Model") # 下拉菜单控件，用于选择模型
                     detect_class = gr.Dropdown(detect_classes, value=0, label="Class", type='index') # 下拉菜单控件，用于选择要检测的目标类别
                     output_dir = gr.Textbox(label="Output dir", value=tempfile.mkdtemp()) # 文本框控件，用于指定输出视频的保存路径，默认为一个临时生成的目录
                     with gr.Row():