Update app.py

app.py

@@ -59,7 +59,8 @@ if st.button('Excute'):
         image_path = check_file(image)
         input_image = Image.open(image_path)
     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=os.path.join('pretrained', pretrained))
+        model = torch.hub.load('ultralytics/yolov5', 'custom', path=pretrained)
     with st.spinner('Updating configuration...'):
         model.conf = float(conf)
         model.max_det = int(max_det)