no message

app.py

@@ -117,24 +117,24 @@ def main():
         
     with CONTAINER_BODY:
 
-        with COL1:
+        with COLS[0]:
             st.markdown("This app uses several 🤗 models to classify images stored in 🤗 datasets.")
             st.write("Soon we will have a dataset template")
 
         #Model
         chosen_model_name = st.selectbox("Select the model to use",  MODELS, index=0)
         if chosen_model_name is not None:
-            COL1.st.write("You selected", chosen_model_name) 
+            COLS[0].st.write("You selected", chosen_model_name) 
 
         #Dataset
         shosen_dataset_name = st.selectbox("Select the dataset to use",  DATASETS, index=0)
         if shosen_dataset_name is not None:
-            COL1.st.write("You selected", shosen_dataset_name)
+            COLS[0].st.write("You selected", shosen_dataset_name)
 
         #click to classify
         #image_object = dataset['pasta'][0]  
         if chosen_model_name is not None and shosen_dataset_name is not None:
-            if COL1.button("Classify images"):
+            if COLS[0].button("Classify images"):
             
                 #classification_array =[]
                 classification_result = classify_full_dataset(shosen_dataset_name, chosen_model_name)