clean

handler.py

@@ -1,14 +1,18 @@
 from typing import Dict, List, Any
-from transformers import LongformerTokenizer
-import torch
-import torch.nn.functional as F
+
 import spacy
 from spacy.matcher import PhraseMatcher
 from skillNer.general_params import SKILL_DB
 from skillNer.skill_extractor_class import SkillExtractor
+
+import torch
+import torch.nn as nn
+import torch.nn.functional as F
+from transformers import LongformerTokenizer, LongformerModel
 import requests
 import os
 from dotenv import load_dotenv
+
 import re
 from datetime import datetime
 import time
@@ -19,6 +23,7 @@ load_dotenv('.env.local')
 Resume_num_labels = None
 class EndpointHandler():
     def __init__(self, path=""):
+        # Label mapping as provided
         # Resume Label Mapping
         self.hf_token = os.getenv('HUGGINGFACE_TOKEN')
         if not self.hf_token: