[Update] Fix issue: fail to load model

app.py

@@ -6,29 +6,6 @@ from yolov7.utils.google_utils import attempt_download_from_hub, attempt_downloa
 from yolov7.utils.torch_utils import TracedModel
 YOLO_MODEL_FILE_NAME="kadirnar/yolov7-v0.1"
 
-def load_local_model(model_file, autoshape=True, device='cpu', trace=False, size=640, half=False, hf_model=False):
-    """
-    Creates a specified YOLOv7 model
-    Arguments:
-        model_path (str): path of the model
-        device (str): select device that model will be loaded (cpu, cuda)
-        trace (bool): if True, model will be traced
-        size (int): size of the input image
-        half (bool): if True, model will be in half precision
-        hf_model (bool): if True, model will be loaded from huggingface hub    
-    Returns:
-        pytorch model
-    (Adapted from yolov7.hubconf.create)
-    """
-    
-    model = attempt_load(model_file, map_location=device)
-    if trace:
-        model = TracedModel(model, device, size)
-    if autoshape:
-        model = autoShape(model)
-    if half:
-        model.half()
-    return model
 
 # YOLO_MODEL_FILE_NAME="kadirnar/yolov7-tiny-v0.1"
 def yolov7_inference(
@@ -37,7 +14,7 @@ def yolov7_inference(
     conf_threshold: gr.inputs.Slider = 0.25,
     iou_threshold: gr.inputs.Slider = 0.45,        
 ):
-    model = yolov7.load_model(YOLO_MODEL_FILE_NAME, device="cpu", hf_model=False, trace=False)
+    model = yolov7.load(YOLO_MODEL_FILE_NAME, device="cpu", hf_model=True, trace=False)
     model.conf = conf_threshold
     model.iou = iou_threshold
     results = model([image], size=image_size)