changed default

pages/1_Textiles_Disentanglement.py

@@ -49,7 +49,7 @@ with dnnlib.util.open_url('./data/textile_model_files/network-snapshot-005000.pk
 COLORS_LIST = ['Gray', 'Red Orange', 'Yellow', 'Green', 'Light Blue', 'Blue', 'Purple', 'Pink']
 
 if 'image_id' not in st.session_state:
-    st.session_state.image_id = 0
+    st.session_state.image_id = 52921
 if 'color_ids' not in st.session_state:
     st.session_state.concept_ids = COLORS_LIST[-1]
 if 'space_id' not in st.session_state:
@@ -139,27 +139,30 @@ with input_col_4:
     with st.form('text_form_2'):
         st.write('Use best options')
         best = st.selectbox('Option:', tuple([True, False]), index=0)
-        st.write('Options for StyleSpace (not available for Saturation and Value)')
-        sign = st.selectbox('Sign option:', tuple([True, False]), index=1)
-        num_factors = st.selectbox('Number of factors option:', tuple([1, 5, 10, 20, False]), index=4)
-        st.write('Options for InterFaceGAN (not available for Saturation and Value)')
-        cl_method = st.selectbox('Classification method option:', tuple(['LR', 'SVM', False]), index=2)
-        regularization = st.selectbox('Regularization option:', tuple([0.1, 1.0, False]), index=2)
-        st.write('Options for InterFaceGAN (only for Saturation and Value)')
-        extremes = st.selectbox('Extremes option:', tuple([True, False]), index=1)
-        
+        if st.session_state.best is False:
+            st.write('Options for StyleSpace (not available for Saturation and Value)')
+            sign = st.selectbox('Sign option:', tuple([True, False]), index=1)
+            num_factors = st.selectbox('Number of factors option:', tuple([1, 5, 10, 20, False]), index=4)
+            st.write('Options for InterFaceGAN (not available for Saturation and Value)')
+            cl_method = st.selectbox('Classification method option:', tuple(['LR', 'SVM', False]), index=2)
+            regularization = st.selectbox('Regularization option:', tuple([0.1, 1.0, False]), index=2)
+            st.write('Options for InterFaceGAN (only for Saturation and Value)')
+            extremes = st.selectbox('Extremes option:', tuple([True, False]), index=1)
+            
         choose_options_button = st.form_submit_button('Choose the defined options')
         # st.write('**Choose a latent space to disentangle**')
         # # chosen_text_id_input = st.empty()
         # # concept_id = chosen_text_id_input.text_input('Concept:', value=st.session_state.concept_id)
         # space_id = st.selectbox('Space:', tuple(['Z', 'W']))
         if choose_options_button:
-            st.session_state.sign = sign
-            st.session_state.num_factors = num_factors
-            st.session_state.cl_method = cl_method
-            st.session_state.regularization = regularization
-            st.session_state.extremes = extremes
             st.session_state.best = best
+            if st.session_state.best is False:
+                st.session_state.sign = sign
+                st.session_state.num_factors = num_factors
+                st.session_state.cl_method = cl_method
+                st.session_state.regularization = regularization
+                st.session_state.extremes = extremes
+                
     
 # with input_col_4:
 #     with st.form('Network specifics:'):