Update app.py

app.py

@@ -37,10 +37,10 @@ def check_file(file, suffix=''):
         return file
 
 def read_pretrain(path):
-    return torch.hub.load('ultralytics/yolov5', 'custom', path=path
+    return torch.hub.load('ultralytics/yolov5', 'custom', path=path)
     
+default_pretrained = '2022.11.04-YOLOv5x6_1280-Hololive_Waifu_Classification.pt'
 with st.spinner('Loading the model...'):
-    default_pretrained = '2022.11.04-YOLOv5x6_1280-Hololive_Waifu_Classification.pt'
     read_pretrain(default_pretrained)
     
 st.title("Hololive Waifu Classification")