Update app.py

app.py

@@ -65,7 +65,7 @@ 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)
 clicked = st.button('Excute')
 with st.spinner('Loading the model...'):
-    read_pretrain(default_pretrained)
+    model = read_pretrain(default_pretrained)
     
 if clicked:
     with st.spinner('Loading the image...'):
@@ -74,7 +74,8 @@ if clicked:
     if default_pretrained != pretrained:
         with st.spinner('Loading the model...'):
             # model = torch.hub.load('ultralytics/yolov5', 'custom', path=os.path.join('pretrained', pretrained))
-            model = torch.hub.load('ultralytics/yolov5', 'custom', path=pretrained)
+            # model = torch.hub.load('ultralytics/yolov5', 'custom', path=pretrained)
+            model = read_pretrain(pretrained)
     with st.spinner('Updating configuration...'):
         model.conf = float(conf)
         model.max_det = int(max_det)