Update handler.py

Author: npatta01

serving/handler.py

@@ -14,7 +14,6 @@
 logger = logging.getLogger(__name__)
 
 from ts.torch_handler.base_handler import BaseHandler
-from captum.attr import LayerIntegratedGradients
 
 from ts.utils.util import map_class_to_label
 
@@ -188,4 +187,4 @@ def handle(self, data, context):
         stop_time = time.time()
         metrics.add_time('HandlerTime', round((stop_time - start_time) * 1000, 2), None, 'ms')
 
-        return data_postprocess
+        return data_postprocess