Updated CLasses display logic

app.py

@@ -201,7 +201,7 @@ def predict(input_image):
     # img = Image.create(input_image)
     pil_image_object = Image.fromarray(input_image)
     probs = process_image(pil_image_object)
-    return {genre_classes[i]: float(probs[i]) for i in range(len(genre_classes)-1)}
+    return {genre_classes[i]: float(probs[i]) for i in range(len(genre_classes))}
 
 image_path_prefx = os.path.join(ROOT_FOLDER,'examples')
 examples = [f"{image_path_prefx}/e{n}.jpeg" for n in range(4)]