Renew model in fine_tune.py

apps/fine_tune.py

@@ -22,7 +22,7 @@ def app():
 
     prediction = st.button("Predict")
     if prediction:
-        ship_model = torch.hub.load('ultralytics/yolov5', 'custom', path="apps/model/main_model.pt", force_reload=True)
+        ship_model = torch.hub.load('ultralytics/yolov5', 'custom', path="apps/model/fine_tune.pt", force_reload=True)
         results = ship_model(f"{folder_path}")
         with st.spinner("Loading..."):
             time.sleep(3.5)