bug fix

app.py

@@ -39,7 +39,7 @@ def classify_full_dataset(shosen_dataset_name, chosen_model_name):
     dataset = load_dataset(shosen_dataset_name,"testedata_readme")
     #Image teste load 
     image_object = dataset['pasta'][0]["image"]
-    st.image(image_object, caption="Uploaded Image", height=300)
+    st.image(image_object, caption="Uploaded Image", width=300)
     st.write("### FLAG 3")
     
     #modle instance
@@ -54,14 +54,10 @@ def classify_full_dataset(shosen_dataset_name, chosen_model_name):
     
     #save classification
     
-    
     image_count += 1
     
     return image_count
 
-
-
-
 def main():
     st.title("Bulk Image Classification DEMO")
     st.markdown("This app uses several 🤗 models to classify images stored in 🤗 datasets.")