iniT

app.py

@@ -28,7 +28,8 @@ with st.sidebar:
         from src.model import Model
         st.session_state.task=task
         model = Model(revision=mapper[task])
-    
+        st.session_state.model = model
+        
 left, middle, right = st.columns([4,1,4])
 
 left_container = left.container(border=True)