Update app.py

app.py

@@ -72,6 +72,9 @@ def deprocess_image(img):
 
 # The visualization function
 def main():
+    # Initialize states
+    initialize_states()
+
     # Model selector
     mn_option = st.selectbox("Select the model for visualization -", AVAILABLE_MODELS)