Update handler.py

handler.py

@@ -114,6 +114,6 @@ class EndpointHandler():
         sorted_ids = torch.argsort(law_token_probs, descending=True)[:topk]
         print([self.law_token_names[x] for x in sorted_ids])
         token_objects = [
-            self.law_lookup.get_law_from_token(self.law_token_names[x]),
+            self.law_lookup.get_law_from_token(self.law_token_names[x])
             for x in sorted_ids.tolist()]
         return {"tokens": token_objects}