testing restart button

app.py

@@ -75,14 +75,17 @@ def main():
     st.title("Bulk Image Classification DEMO")
     col1, col2 = st.columns([3, 1])
     
+    # Restart or reset your app
+    if st.button("Restart"):
+    # Code to restart or reset your app goes here
+    st.experimental_rerun()
+    
+    
     with col1:
         st.markdown("This app uses several 🤗 models to classify images stored in 🤗 datasets.")
         st.write("Soon we will have a dataset template")
         
-        ## Restart or reset your app
-        #if st.button("Restart"):
-        ## Code to restart or reset your app goes here
-        #    st.experimental_rerun()
+      
             
         #Model
         chosen_model_name = st.selectbox("Select the model to use",  MODELS, index=0)