Update visualization.py

visualization.py

@@ -61,6 +61,7 @@ def print_top_classes(original_image, **kwargs):
     class_indices = predictions.data.topk(5, dim=1)[1][0].tolist()
     max_str_len = 0
     class_names = []
+    output = []
     for cls_idx in class_indices:
         class_names.append(CLS2IDX[cls_idx])
         if len(CLS2IDX[cls_idx]) > max_str_len: