fixed the output dict error

app.py

@@ -47,12 +47,9 @@ def infer(input_image):
     image_tensor = read_image(input_image)
     print(image_tensor.shape)
     predictions = model.predict(np.expand_dims((image_tensor), axis=0))
-    predictions = np.squeeze(predictions)
-    predictions = np.argmax(predictions) # , axis=2
-    predicted_label = class_names[predictions.item()]
-    return str(predicted_label)
+    predictions = np.squeeze(predictions).astype(float)
 
-    # return dict(zip(class_names, predictions))
+    return dict(zip(class_names, predictions))
     
     
 # get the inputs