earlsab commited on
Commit
8b1d0a4
·
1 Parent(s): ea32026
Files changed (1) hide show
  1. handler.py +1 -10
handler.py CHANGED
@@ -1,22 +1,14 @@
1
  from typing import Dict, List, Any
2
- from transformers import AutoModelForQuestionAnswering, AutoTokenizer, pipeline, LongformerTokenizer
3
  import torch
4
- import torch.nn as nn
5
  import torch.nn.functional as F
6
  import spacy
7
  from spacy.matcher import PhraseMatcher
8
- from transformers import LongformerModel
9
  from skillNer.general_params import SKILL_DB
10
  from skillNer.skill_extractor_class import SkillExtractor
11
- import torch
12
- from transformers import LongformerTokenizer
13
  import requests
14
  import os
15
  from dotenv import load_dotenv
16
-
17
- import torch
18
- import torch.nn.functional as F
19
- from transformers import LongformerTokenizer
20
  import re
21
  from datetime import datetime
22
  import time
@@ -27,7 +19,6 @@ load_dotenv('.env.local')
27
  Resume_num_labels = None
28
  class EndpointHandler():
29
  def __init__(self, path=""):
30
- # Label mapping as provided
31
  # Resume Label Mapping
32
  self.hf_token = os.getenv('HUGGINGFACE_TOKEN')
33
  if not self.hf_token:
 
1
  from typing import Dict, List, Any
2
+ from transformers import LongformerTokenizer
3
  import torch
 
4
  import torch.nn.functional as F
5
  import spacy
6
  from spacy.matcher import PhraseMatcher
 
7
  from skillNer.general_params import SKILL_DB
8
  from skillNer.skill_extractor_class import SkillExtractor
 
 
9
  import requests
10
  import os
11
  from dotenv import load_dotenv
 
 
 
 
12
  import re
13
  from datetime import datetime
14
  import time
 
19
  Resume_num_labels = None
20
  class EndpointHandler():
21
  def __init__(self, path=""):
 
22
  # Resume Label Mapping
23
  self.hf_token = os.getenv('HUGGINGFACE_TOKEN')
24
  if not self.hf_token: