Update app.py

app.py

@@ -3,7 +3,7 @@ import tensorflow as tf
 import numpy as np
 from numpy import asarray
 from datetime import datetime
-
+import cv2
 
 model = tf.keras.models.load_model("simple-CNN-model.2022-8-9.hdf5")
 
@@ -18,7 +18,10 @@ def image_predict(img):
     global model
     if model is None:
         model = tf.keras.models.load_model("models/simple-CNN-model.2022-8-9.hdf5")
-        
+    
+    #img = img.reshape((None, 227, 227, 3)) 
+    
+    img = cv2.resize(img, (227, 227))   
     data = np.asarray(img)
         
     ndata = np.expand_dims(data, axis=0)
@@ -53,14 +56,8 @@ iface = gr.Interface(
     description = "App classifying objects on different colors",
     article = "<p style='text-align: center'><a href='https://www.rrighart.com' target='_blank'>Webpage</a></p>",
     fn=image_predict,
-    inputs=gr.Image(shape=(227,227)), 
+    inputs=gr.Image(width=227, height=227), # shape=(227,227) 
     outputs=gr.Label(),
     examples=['shoes1.jpg', 'shoes2.jpg'],
-    enable_queue=True,
-    interpretation="default",
-    debug=True
     )
 iface.launch()
-
-
-