Update app.py

app.py

@@ -23,7 +23,7 @@ from huggingface_hub import hf_hub_download
 
 def load_model(model_name):
     if model_name == 'yolov7-DSSLab':
-        model_path = hf_hub_download(repo_id=f"llmx/yolov7_det_weapon", filename=f"best_v1.pt")
+        model_path = hf_hub_download(repo_id=f"llmx/yolov7_det_weapon", filename=f"best_v1.pt", token=os.environ['Huggingface-TOKEN'])
     else:
         model_path = hf_hub_download(repo_id=f"Yolov7/{model_name}", filename=f"{model_name}.pt")
 
@@ -197,5 +197,9 @@ 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-DSSLab",examples=[["horses.jpeg", "yolov7-DSSLab"]],description="经过额外训练的Yolov7 Demo").launch()
+imgs = os.listdir('./Inference/examples/')
+examples = []
+for img in imgs:
+   examples.append([f"./Inference/examples/{img}","yolov7-DSSLab"])
+
+gr.Interface(detect,[gr.Image(type="pil"),gr.Dropdown(choices=model_names)], gr.Image(type="pil"),title="Yolov7-DSSLab",examples=examples,description="在yolov7基础上经过额外训练的Demo").launch()