done

app.py

@@ -22,34 +22,36 @@ SCROLLABLE_TEXT = COLS[1].container(height=500)
 def classify_full_dataset(shosen_dataset_name, chosen_model_name):
     image_count = 0
      
-     
+    #modle instance
+    classifier_pipeline = pipeline('image-classification', model=chosen_model_name)    
+        
+    #dataset
+    dataset = load_dataset(shosen_dataset_name,"testedata_readme") 
      
     for i in range(len(dataset)):
         image_object = dataset['pasta'][i]["image"]
         SCROLLABLE_TEXT.image(image_object, caption="Uploaded Image", width=300)
-        
-     
-     
-    #dataset
-    dataset = load_dataset(shosen_dataset_name,"testedata_readme")
+        #classification
+        classification_result = classifier_pipeline(image_object)
+        SCROLLABLE_TEXT.write(classification_result)
+        #TODO save classification result in dataset
+        image_count += 1
+        SCROLLABLE_TEXT.write(f"Image count" + str(image_count))
+        #SCROLLABLE_TEXT.write(image_count)    
+
  
     #Image teste load 
-    image_object = dataset['pasta'][0]["image"]
-    SCROLLABLE_TEXT.image(image_object, caption="Uploaded Image", width=300)
+    #image_object = dataset['pasta'][0]["image"]
+    #SCROLLABLE_TEXT.image(image_object, caption="Uploaded Image", width=300)
+
 
-    #modle instance
-    classifier_pipeline = pipeline('image-classification', model=chosen_model_name)
     
-    #classification
-    classification_result = classifier_pipeline(image_object)
-    SCROLLABLE_TEXT.write(classification_result)
+    
 
     
-    #TODO save classification
     
-    image_count += 1
-    SCROLLABLE_TEXT.write("Image count")
-    SCROLLABLE_TEXT.write(image_count)  
+    
+
 
 def main():