no message

app.py

@@ -15,24 +15,11 @@ DATASETS = [
 MAX_N_LABELS = 5
 SPLIT_TO_CLASSIFY = 'pasta'
 
-# COL1, COL2 = st.columns([3, 1])
-# CONTAINER_TOP = st.container()
-# CONTAINER_BODY = st.container()
-# CONTAINER_FULL = st.container()
-# CONTAINER_LOOP = st.container()
-COL1=""        
-COL2="" 
 COLS = st.columns([3, 1]) 
-CONTAINER_TOP = st.container() 
-CONTAINER_BODY = st.container()
-CONTAINER_FULL = st.container() 
-CONTAINER_LOOP = st.container() 
 
 
 
-#(image_object, classifier_pipeline)
-#def classify_one_image(classifier_model, dataset_to_classify):
-#classify_one_image(image_object, classifier_pipeline)
+
 def classify_one_image(classifier_model, dataset_to_classify):
        
        
@@ -103,17 +90,8 @@ def make_template():
 
 def main():
     
-    make_template()
-    
-    # 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
-        #     import subprocess
-        #     subprocess.call(["shutdown", "-r", "-t", "0"])
+     
+    COLS[0].write("# Bulk Image Classification App")
         
     #with CONTAINER_BODY:
     with COLS[0]:
@@ -121,14 +99,14 @@ def main():
         st.write("Soon we will have a dataset template")
         
     #Model
-    chosen_model_name = COLS[0].selectbox("Select the model to use",  MODELS, index=0)
+    chosen_model_name = COLS[0].selectbox(f"Select the model to use",  MODELS, index=0)
     if chosen_model_name is not None:
-        COLS[0].write("You selected", chosen_model_name) 
+        COLS[0].write(f"You selected", chosen_model_name) 
         
     #Dataset
-    shosen_dataset_name = COLS[0].selectbox("Select the dataset to use",  DATASETS, index=0)
+    shosen_dataset_name = COLS[0].selectbox(f"Select the dataset to use",  DATASETS, index=0)
     if shosen_dataset_name is not None:
-        COLS[0].write("You selected", shosen_dataset_name)
+        COLS[0].write(f"You selected", shosen_dataset_name)
         
     #click to classify
     #image_object = dataset['pasta'][0]  
@@ -140,6 +118,7 @@ def main():
             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()