Update src/streamlit_app.py

src/streamlit_app.py

@@ -165,6 +165,8 @@ if 'model' not in st.session_state:
 
     st.session_state.current_prediction = None
 
+    st.session_state.label_input = None
+
 
 def fetch_image():
 
@@ -212,7 +214,7 @@ if st.session_state.current_image is not None:
     prediction = st.session_state.model.predict(np.array([st.session_state.current_image]))[0][0]
 
     st.session_state.current_prediction = 'dog' if prediction > 0.5 else 'cat'
-    st.success(f"**Model Predicts:** {st.session_state.current_prediction} (cat-confidence {(1-prediction)*100:.2f}%; dog-confidence {(prediction)*100:.2f}%")
+    st.success(f"**Model Predicts:** {st.session_state.current_prediction} --- (cat-confidence {(1-prediction)*100:.2f}%; dog-confidence {(prediction)*100:.2f}%)")
 
     st.image(st.session_state.unprocessed_image, caption="Current Image")