bills commited on
Commit
967af65
1 Parent(s): 580a94d

Add another change on app.py

Browse files
Files changed (1) hide show
  1. apps.py +2 -1
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}")