Spaces:
Runtime error
Runtime error
bills
commited on
Commit
•
967af65
1
Parent(s):
580a94d
Add another change on app.py
Browse files
apps.py
CHANGED
@@ -59,11 +59,12 @@ eval_col4.metric("mAP 0.5:0.95", "62.63%")
|
|
59 |
uploaded_file = st.file_uploader("Choose a ship imagery")
|
60 |
if uploaded_file is not None:
|
61 |
st.image(uploaded_file, caption='Image to predict')
|
|
|
|
|
62 |
# st.write(uploaded_file.)
|
63 |
|
64 |
prediction = st.button("Predict")
|
65 |
if prediction:
|
66 |
-
folder_path = st.text_input('Input the folder path of image')
|
67 |
ship_model = torch.hub.load('ultralytics/yolov5', 'custom', path="supercomputer/best.pt", force_reload=True)
|
68 |
# results = ship_model(f"C:/Users/bilva/YOLOv5/ship_test/{uploaded_file.name}")
|
69 |
results = ship_model(f"{folder_path}/{uploaded_file.name}")
|
|
|
59 |
uploaded_file = st.file_uploader("Choose a ship imagery")
|
60 |
if uploaded_file is not None:
|
61 |
st.image(uploaded_file, caption='Image to predict')
|
62 |
+
folder_path = st.text_input('Input the folder path of image')
|
63 |
+
|
64 |
# st.write(uploaded_file.)
|
65 |
|
66 |
prediction = st.button("Predict")
|
67 |
if prediction:
|
|
|
68 |
ship_model = torch.hub.load('ultralytics/yolov5', 'custom', path="supercomputer/best.pt", force_reload=True)
|
69 |
# results = ship_model(f"C:/Users/bilva/YOLOv5/ship_test/{uploaded_file.name}")
|
70 |
results = ship_model(f"{folder_path}/{uploaded_file.name}")
|