Spaces:
Runtime error
Runtime error
bills
commited on
Commit
•
56f6107
1
Parent(s):
b66a7d9
Renew model in fine_tune.py
Browse files- apps/fine_tune.py +1 -1
apps/fine_tune.py
CHANGED
@@ -22,7 +22,7 @@ def app():
|
|
22 |
|
23 |
prediction = st.button("Predict")
|
24 |
if prediction:
|
25 |
-
ship_model = torch.hub.load('ultralytics/yolov5', 'custom', path="apps/model/
|
26 |
results = ship_model(f"{folder_path}")
|
27 |
with st.spinner("Loading..."):
|
28 |
time.sleep(3.5)
|
|
|
22 |
|
23 |
prediction = st.button("Predict")
|
24 |
if prediction:
|
25 |
+
ship_model = torch.hub.load('ultralytics/yolov5', 'custom', path="apps/model/fine_tune.pt", force_reload=True)
|
26 |
results = ship_model(f"{folder_path}")
|
27 |
with st.spinner("Loading..."):
|
28 |
time.sleep(3.5)
|