no message

app.py

@@ -64,7 +64,7 @@ def classify_full_dataset(shosen_dataset_name, chosen_model_name):
      
     #dataset
     dataset = load_dataset(shosen_dataset_name,"testedata_readme")
-    with CONTAINER_LOOP:
+    with COLS[1]:
         #Image teste load 
         image_object = dataset['pasta'][0]["image"]
         st.image(image_object, caption="Uploaded Image", width=300)
@@ -72,18 +72,18 @@ def classify_full_dataset(shosen_dataset_name, chosen_model_name):
         
     #modle instance
     classifier_pipeline = pipeline('image-classification', model=chosen_model_name)
-    CONTAINER_LOOP.write("### FLAG 4")
+    COLS[1].write("### FLAG 4")
     
     #classification
     classification_result = classifier_pipeline(image_object)
-    CONTAINER_LOOP.write(classification_result)
-    CONTAINER_LOOP.write("### FLAG 5")
+    COLS[1].write(classification_result)
+    COLS[1].write("### FLAG 5")
     #classification_array.append(classification_result)
     
     #save classification
     
     image_count += 1
-    CONTAINER_LOOP.write(f"Image count: {image_count}")
+    COLS[1].write(f"Image count: {image_count}")
     
     return image_count
 
@@ -105,10 +105,10 @@ def main():
     
     make_template()
     
-    with CONTAINER_TOP:
-        st.write("# Bulk Image Classification DEMO")
+    # with CONTAINER_TOP:
+    #     st.write("# Bulk Image Classification DEMO")
 
-    
+    COLS[0].write("# Bulk Image Classification DEMO")
     # TODO Restart or reset your app
     # if st.button("Restart"):
     # # Code to restart or reset your app goes here
@@ -116,7 +116,7 @@ def main():
         #     subprocess.call(["shutdown", "-r", "-t", "0"])
         
     #with CONTAINER_BODY:
-    with CONTAINER_TOP:
+    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")
         
@@ -137,9 +137,9 @@ def main():
         
             #classification_array =[]
             classification_result = classify_full_dataset(shosen_dataset_name, chosen_model_name)
-            CONTAINER_LOOP.write(f"Classification result: {classification_result}")
+            COLS[0].write(f"Classification result: {classification_result}")
             #classification_array.append(classification_result)
             #st.write("# FLAG 6")
-            #st.write(classification_array)   
+            #st.write(classification_array)       
 if __name__ == "__main__":
     main()