Fixing input shape

app.py

@@ -15,7 +15,9 @@ labels = [x for x in response.text.split("\n") if x != ""]
 model = pull_from_hub(repo_id="Dimitre/mobilenet_v3_small")
 
 def preprocess(image):
-    image = image.reshape((-1, 224, 224, 3))
+    print(image.shape)
+    # image = image.reshape((-1, 224, 224, 3))
+    # print(image.shape)
     return image / 255.
 
 def postprocess(prediction):
@@ -23,7 +25,7 @@ def postprocess(prediction):
 
 def predict_fn(image):
     image = preprocess(image)
-    prediction = model(image)
+    prediction = model([image])
     scores = postprocess(prediction)
     return scores