debug input shape

app.py

@@ -16,21 +16,22 @@ model = pull_from_hub(repo_id="Dimitre/mobilenet_v3_small")
 
 def preprocess(image):
     print(image.shape)
-    # image = image.reshape((-1, 224, 224, 3))
-    # print(image.shape)
+    image = image.reshape((-1, 224, 224, 3)) # (batch_size, height, width, num_channels)
+    print(image.shape)
     return image / 255.
 
 def postprocess(prediction):
-    return {labels[i]: prediction[i] for i in range(len(labels))}
+    # return {labels[i]: prediction[i] for i in range(len(labels))}
+    return {labels[i]: 0 for i in range(len(labels))}
 
 def predict_fn(image):
     image = preprocess(image)
-    prediction = model([image])
+    prediction = model(image)
+    print(prediction)
     scores = postprocess(prediction)
     return scores
 
 iface = gr.Interface(fn=predict_fn, 
                      inputs=gr.Image(shape=(224, 224)), 
-                     outputs=gr.Label(num_top_classes=5), 
-                     examples=["apples.jpeg", "banana.jpeg", "car.jpeg"])
+                     outputs=gr.Label(num_top_classes=5))
 iface.launch()