Update app.py

app.py

@@ -12,15 +12,9 @@ uploaded_img = st.file_uploader("Upload your file here...",type=['png', 'jpeg',
 
 if uploaded_img is not None:
     st.image(uploaded_img)
-    img = Image.open(uploaded_img)
-    print(img.mode)
-    resized_img = img.resize((160,160))
-    print(resized_img.mode)
-    input_arr = keras.preprocessing.image.img_to_array(resized_img)
-    print(input_arr.shape)
-    input_arr = input_arr.astype('float32') / 255.
-    print(input_arr.shape)
-    result = loaded_model.predict(input_arr)
+    img = Image.open(uploaded_img).resize((160, 160), Image.ANTIALIAS)
+    img = np.array(img)
+    result = model.predict(img[None,:,:])
     st.write(f"Your prediction is: {result}")