Update app.py

app.py

@@ -18,7 +18,7 @@ base_model.trainable = False
 
 def img_pros(img):
   original_img = tf.keras.preprocessing.image.img_to_array(img)
-  original_img_shape = img.shape
+  original_img_shape = original_img.shape
   img = tf.image.resize(original_img, [160,160])
   img = tf.expand_dims(img, axis = 0)
   return img, original_img, original_img_shape