Update app.py

app.py

@@ -3,10 +3,13 @@ from transformers import pipeline
 from transformers import ViTFeatureExtractor, ViTForImageClassification
 from PIL import Image
 import requests
+import numpy as np
 
 img_file_buffer = st.file_uploader('Upload a PNG image', type='png')
 if img_file_buffer is not None:
-    image = Image.open(img_file_buffer)    
+    image = Image.open(img_file_buffer)  
+    image_array=np.array(image) 
+    print("Image_array.shape:",image_array.shape)
     
     feature_extractor = ViTFeatureExtractor.from_pretrained('rizvandwiki/gender-classification')
     model = ViTForImageClassification.from_pretrained('rizvandwiki/gender-classification')