Update app.py

app.py

@@ -37,6 +37,10 @@ model = load_learner("model.pkl")
 
 def predict(img):
     logging.warning(f"{img}")
+    try:
+        logger.warning(f"Shape: {img.shape}")
+    except AttributeError:
+        logger.warning(f"Size: {img.size}")
     pred, idx, probs = model.predict(img)
     logging.debug(f"{type(probs)}")
     logging.debug(f"{type(probs[0])}")