Update app.py

app.py

@@ -20,6 +20,7 @@ from PIL import Image
  
 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")
 
@@ -32,10 +33,12 @@ model_names = [
     "yolov7-e6",
 ]
 
+models = {model_name: load_model(model_name) for model_name in model_names}
+
 
 def detect(img,model):
     parser = argparse.ArgumentParser()
-    parser.add_argument('--weights', nargs='+', type=str, default=load_model(model), help='model.pt path(s)')
+    parser.add_argument('--weights', nargs='+', type=str, default=models[model], help='model.pt path(s)')
     parser.add_argument('--source', type=str, default='Inference/', help='source')  # file/folder, 0 for webcam
     parser.add_argument('--img-size', type=int, default=640, help='inference size (pixels)')
     parser.add_argument('--conf-thres', type=float, default=0.25, help='object confidence threshold')