Update app.py

app.py

@@ -40,8 +40,6 @@ def read_pretrain(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...'):
-    read_pretrain(default_pretrained)
     
 st.title("Hololive Waifu Classification")
 
@@ -65,7 +63,9 @@ multi_label = st.selectbox('Multiple labels per box', (False, True))
 agnostic = st.selectbox('Class-agnostic', (False, True))
 amp = st.selectbox('Automatic Mixed Precision inference', (False, True))
 max_det  = st.number_input(label='Maximum number of detections per image', min_value=None, max_value=None, value=1000, step=1)
-
+with st.spinner('Loading the model...'):
+    read_pretrain(default_pretrained)
+    
 if st.button('Excute'):
     with st.spinner('Loading the image...'):
         image_path = check_file(image)