Add predicted objects to predict command

Closes #36 (closed)

dan/predict/attention.py

@@ -3,7 +3,6 @@ import re
 
 import cv2
 import numpy as np
-import math
 from PIL import Image
 
 from dan import logger
@@ -74,8 +73,8 @@ def plot_attention(
     :param scale: Scaling factor for the output gif image
     :param outname: Name of the gif image
     """
-    
-    height, width, _ = image.shape 
+
+    height, width, _ = image.shape
     attention_map = []
 
     # Convert to PIL Image and create mask