no message

app.py

@@ -115,33 +115,31 @@ def main():
         #     import subprocess
         #     subprocess.call(["shutdown", "-r", "-t", "0"])
         
-    with CONTAINER_BODY:
-
-        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:
-            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:
-            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 COLS[0].button("Classify images"):
-            
-                #classification_array =[]
-                classification_result = classify_full_dataset(shosen_dataset_name, chosen_model_name)
-                CONTAINER_LOOP.write(f"Classification result: {classification_result}")
-                #classification_array.append(classification_result)
-                #st.write("# FLAG 6")
-                #st.write(classification_array)   
-
+    #with CONTAINER_BODY:
+    with CONTAINER_TOP:
+        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 = COLS[0].selectbox("Select the model to use",  MODELS, index=0)
+    if chosen_model_name is not None:
+        COLS[0].write("You selected", chosen_model_name) 
+        
+    #Dataset
+    shosen_dataset_name = COLS[0].selectbox("Select the dataset to use",  DATASETS, index=0)
+    if shosen_dataset_name is not None:
+        COLS[0].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 COLS[0].button("Classify images"):
+        
+            #classification_array =[]
+            classification_result = classify_full_dataset(shosen_dataset_name, chosen_model_name)
+            CONTAINER_LOOP.write(f"Classification result: {classification_result}")
+            #classification_array.append(classification_result)
+            #st.write("# FLAG 6")
+            #st.write(classification_array)   
 if __name__ == "__main__":
     main()