Update app.py

app.py

@@ -22,12 +22,16 @@ from huggingface_hub import hf_hub_download
 
 
 def load_model(model_name):
-    model_path = hf_hub_download(repo_id=f"Yolov7/{model_name}", filename=f"{model_name}.pt")
+    if model_name == 'yolov7-DSSLab':
+        model_path = hf_hub_download(repo_id=f"llmx/yolov7_det_weapon", filename=f"best_v1.pt")
+    else:
+        model_path = hf_hub_download(repo_id=f"Yolov7/{model_name}", filename=f"{model_name}.pt")
 
     return model_path
 
 
 model_names = [
+    "yolov7-DSSLab",
     "yolov7",
     "yolov7-e6e",
     "yolov7-e6",
@@ -194,4 +198,4 @@ def detect(img,model):
     return Image.fromarray(im0[:,:,::-1])
 
    
-gr.Interface(detect,[gr.Image(type="pil"),gr.Dropdown(choices=model_names)], gr.Image(type="pil"),title="Yolov7",examples=[["horses.jpeg", "yolov7"]],description="demo for <a href='https://github.com/WongKinYiu/yolov7' style='text-decoration: underline' target='_blank'>WongKinYiu/yolov7</a> Trainable bag-of-freebies sets new state-of-the-art for real-time object detectors").launch()
+gr.Interface(detect,[gr.Image(type="pil"),gr.Dropdown(choices=model_names)], gr.Image(type="pil"),title="Yolov7-DSSLab",examples=[["horses.jpeg", "yolov7-DSSLab"]],description="经过额外训练的Yolov7 Demo").launch()