no message

app.py

@@ -14,8 +14,9 @@ DATASETS = [
 ]
 MAX_N_LABELS = 5
 
-
-def old_classify_images(classifier_model, dataset_to_classify):
+#(image_object, classifier_pipeline)
+#def classify_one_image(classifier_model, dataset_to_classify):
+def classify_one_image(classifier_model, dataset_to_classify):
        
     for image in dataset:
         st("Image classification: ", image['file'])
@@ -32,28 +33,31 @@ def old_classify_images(classifier_model, dataset_to_classify):
 
 
 def classify_full_dataset(shosen_dataset_name, chosen_model_name):
-    
+    image_count = 0
+     
     #dataset
     dataset = load_dataset(shosen_dataset_name,"testedata_readme")
-    st.write("# FLAG 2")
-    
     #Image teste load 
     image_object = dataset['pasta'][0]["image"]
-    st.image(image_object, caption="Uploaded Image", use_column_width=True)
-    st.write("# FLAG 3")
+    st.image(image_object, caption="Uploaded Image", height=300)
+    st.write("### FLAG 3")
     
     #modle instance
     classifier_pipeline = pipeline('image-classification', model=chosen_model_name)
-    #classifier_pipeline = pipeline('image-classification', model="nateraw/vit-age-classifier", device=0)
-    st.write("# FLAG 4")
+    st.write("### FLAG 4")
+    
     #classification
-    classification_result = classify(image_object, classifier_pipeline)
+    classification_result = classify_one_image(image_object, classifier_pipeline)
     st.write(classification_result)
-    st.write("# FLAG 5")
-
-      
+    st.write("### FLAG 5")
+    #classification_array.append(classification_result)
     
-    return "done"
+    #save classification
+    
+    
+    image_count += 1
+    
+    return image_count
 
 
 
@@ -77,15 +81,13 @@ def main():
     #image_object = dataset['pasta'][0]  
     if chosen_model_name is not None and shosen_dataset_name is not None:
         if st.button("Classify images"):
-            
-            classification_array =[]
-            
-            st.write("# FLAG 1")
-            classification_result = classify(shosen_dataset_name, chosen_model_name)
-            
-            classification_array.append(classification_result)
-            st.write("# FLAG 6")
-            st.write(classification_array)   
+           
+            #classification_array =[]
+            classification_result = classify_full_dataset(shosen_dataset_name, chosen_model_name)
+            st.write(f"Classification result: {classification_result}")
+            #classification_array.append(classification_result)
+            #st.write("# FLAG 6")
+            #st.write(classification_array)   
 
 if __name__ == "__main__":
     main()