no message

app.py

@@ -13,21 +13,33 @@ DATASETS = [
             "Nunt/backup_leonardo_2024-02-01" 
 ]
 MAX_N_LABELS = 5
+SPLIT_TO_CLASSIFY = 'pasta'
 
 #(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):
        
-    for image in dataset:
-        st("Image classification: ", image['file'])
-    '''
-        image_path = image['file']
-        img = Image.open(image_path)
-        st.image(img, caption="Original image", use_column_width=True)
-        results = classifier(image_path, top_k=MAX_N_LABELS)
-        st.write(results)
-        st.write("----")
-    '''
+       
+          
+    #image_object = dataset[SPLIT_TO_CLASSIFY][i]["image"]
+    #st.image(image_object, caption="Uploaded Image", width=300)
+
+       
+    #for i in range(len(dataset_to_classify)): 
+    #for image in dataset_to_classify:
+        #image_object = dataset[SPLIT_TO_CLASSIFY][i]["image"]
+        #st.image(image_object, caption="Uploaded Image", width=300)
+        
+        #st.write(f"Image classification: ", image['file'])
+
+        # image_path = image['file']
+        # img = Image.open(image_path)
+        # st.image(img, caption="Original image", use_column_width=True)
+        # results = classifier(image_path, top_k=MAX_N_LABELS)
+        # st.write(results)
+        # st.write("----")
+    
     return "done" 
 
 
@@ -47,7 +59,7 @@ def classify_full_dataset(shosen_dataset_name, chosen_model_name):
     st.write("### FLAG 4")
     
     #classification
-    classification_result = classify_one_image(image_object, classifier_pipeline)
+    classification_result = classifier_pipeline(image_object)
     st.write(classification_result)
     st.write("### FLAG 5")
     #classification_array.append(classification_result)