Add new package

apps.py

@@ -56,13 +56,14 @@ eval_col2.metric("Recall", "83.54%")
 eval_col3.metric("mAP 0.5", "85.39%")
 eval_col4.metric("mAP 0.5:0.95", "62.63%")
 
+st.write('Input the folder path of image')
+
 uploaded_file = st.file_uploader("Choose a ship imagery")
 if uploaded_file is not None:
     st.image(uploaded_file, caption='Image to predict')
 
-folder_path = st.text_input('Input the folder path of image')
 
-    # st.write(uploaded_file.)
+    st.write(uploaded_file.name)
 
 prediction = st.button("Predict")
 if prediction:

requirements.txt

@@ -14,4 +14,5 @@ opencv-contrib-python==4.5.4.58
 folium==0.12.1
 streamlit-folium
 ipython==7.28.0
-psutil==5.9.1
+psutil==5.9.1
+torchvision==0.13.0