testing old label

app.py

@@ -48,11 +48,11 @@ def infer(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.argmax(predictions) # , axis=2
+    predicted_label = class_names[predictions.item()]
+    return str(predicted_label)
 
-    return dict(zip(class_names, predictions))
+    # return dict(zip(class_names, predictions))
     
     
 # get the inputs