From 9ab754b40d7aca0b27e070dd733d570053e38ae5 Mon Sep 17 00:00:00 2001
From: Marco Tulio Ribeiro <marcotcr@gmail.com>
Date: Tue, 11 Jul 2017 17:27:49 -0300
Subject: [PATCH] Fixed bug in lime_image where top_labels=None

---
 lime/lime_image.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/lime/lime_image.py b/lime/lime_image.py
index 72dce4e3..c80ee155 100644
--- a/lime/lime_image.py
+++ b/lime/lime_image.py
@@ -170,6 +170,8 @@ def explain_instance(self, image, classifier_fn, labels=(1,),
         else:
             fudged_image[:] = hide_color
 
+        top = labels
+
         data, labels = self.data_labels(image, fudged_image, segments,
                                         classifier_fn, num_samples,
                                         batch_size=batch_size)