Update app.py

app.py

@@ -4,6 +4,7 @@ from transformers import pipeline
 from huggingface_hub import from_pretrained_keras
 import keras
 import numpy as np
+from PIL import Image
 
 loaded_model = keras.saving.load_model("best_model.keras")
 
@@ -11,7 +12,9 @@ uploaded_img = st.file_uploader("Upload your file here...",type=['png', 'jpeg',
 
 if uploaded_img is not None:
     st.image(uploaded_img)
-    array = np.array(uploaded_img)
+    img = Image.open(uploaded_img)
+    resized_image = image.resize((320,320))
+    array = np.array(resized_image)
     result = loaded_model.predict(array)
     st.write(f"Your prediction is: {result}")