no message

app.py

@@ -27,6 +27,7 @@ def classify_images(classifier_model, dataset_to_classify):
         st.write(results)
         st.write("----")
     '''
+    return "done"
 
 
 
@@ -54,13 +55,22 @@ def main():
            
     '''
         click to classify
+        image_object = dataset['pasta'][0]
     '''    
     if shosen_model_name is not None and shosen_dataset_name is not None:
         if st.button("Classify images"):
+            st.write("# FLAG 1")
+            
             dataset = load_dataset(shosen_dataset_name,"testedata_readme")
-            st.image(dataset[0], caption="Uploaded Image", use_column_width=True)
+            st.write("# FLAG 2")
+            
+            st.image(dataset['pasta'][0], caption="Uploaded Image", use_column_width=True)
+            #st.image(dataset[0], caption="Uploaded Image", use_column_width=True)
+            st.write("# FLAG 3")
+            
             classifier = pipeline('image-classification', model=shosen_model_name, device=0)
-            classify_images(classifier, dataset)
+            st.write("# FLAG 4")
+