Update handler.py

handler.py

@@ -1,7 +1,6 @@
 from typing import Dict, List, Any
-from transformers import AutoModelForTokenClassification
 
-from transformers import AutoProcessor
+from transformers import AutoModelForTokenClassification, AutoProcessor
 import torch
 from subprocess import run
 
@@ -24,8 +23,6 @@ device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 class EndpointHandler:
     def __init__(self, path=""):
         # load model and processor from path
-        
- 
         self.model = AutoModelForTokenClassification.from_pretrained(path).to(device)
         self.processor = AutoProcessor.from_pretrained(path)
 
@@ -63,4 +60,5 @@ class EndpointHandler:
             text = self.processor.tokenizer.decode(inp_ids)
             bbox = unnormalize_box(bbox.tolist(), image.width, image.height)
             result.append({"label": label, "score": score, "text": text, "bbox": bbox})
-        return {"predictions": result}
+        return {"predictions": result}
+