modified output

app.py

@@ -71,7 +71,7 @@ def segmentation_generation(image_name, c_width, n_top, n_gif_imgs):
                                               verbose=True)
     rout = exp_init.explain(classifier_f=cur_model,
                             data=np.ones_like(xtrain[0]),
-                            label=int(prediction),
+                            label=int(prediction[0]),
                             cred_width=c_width,
                             focus_sample=False,
                             l2=False)

image_posterior.py

@@ -96,7 +96,7 @@ def create_gif(explanation_blr, img_name, segments, image, prediction, n_images=
         + "</div>"
     )
 
-    return html, f"Predction was {prediction}: {labels_dict[prediction]}"
+    return html, f"Predction was {prediction}: {labels_dict[str(prediction)]}"