Update handler.py

handler.py

@@ -22,7 +22,7 @@ class EndpointHandler():
 
         for index, input_string in enumerate(inputs):
             if input_string in self.cache:
-                cached_results[index] = self.cache[input_string]
+                cached_results[index] = self.cache[input_string]["pooler_output"]
             else:
                 uncached_inputs.append((index, input_string))