Update app.py

app.py

@@ -36,8 +36,8 @@ def check_file(file, suffix=''):
             ).st_size > 0, f'File download failed: {url}'  # check
         return file
 
-# def read_pretrain(path):
-#     return torch.hub.load('ultralytics/yolov5', 'custom', path=path)
+def read_pretrain(path):
+    return torch.hub.load('ultralytics/yolov5', 'custom', path=path)
     
 # default_pretrained = '2022.11.04-YOLOv5x6_1280-Hololive_Waifu_Classification.pt'