upload

jagadeshchilla · Oct 17, 2024 · a8a41c4 · a8a41c4
1 parent cf2ad5f
commit a8a41c4
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/pages/Image_prediction.py b/pages/Image_prediction.py
@@ -105,7 +105,10 @@ def show_image_prediction():
     # Initialize session state
     if 'saved_predictions' not in st.session_state:
         st.session_state.saved_predictions = []
-
+    if 'predictions' not in st.session_state:
+        st.session_state.predictions = []
+    if 'uploaded_images' not in st.session_state:
+        st.session_state.uploaded_images = []
     # Model selection
     model_selection = st.selectbox("Select a model", list(model_links.keys()))