updates

app.py

@@ -149,10 +149,12 @@ with col1:
   with scol1:
     st.button('Change Image', on_click=button_click)
 
-    st.session_state["model"] = st.selectbox(
+    model = st.selectbox(
     'Model',tuple(pred_dict.keys()))
 
-    st.write(st.session_state["model"])
+    st.session_state["model"]=model
+
+    st.write(model)
 
   with scol2: 
     st.button('Find similar parts', on_click=find)