Vaibhav84 commited on
Commit
2885677
·
1 Parent(s): 957a3f6
Files changed (1) hide show
  1. app.py +9 -0
app.py CHANGED
@@ -7,6 +7,7 @@ import json
7
  import psycopg2
8
  import pandas as pd
9
  import numpy as np
 
10
  import spacy
11
  from sklearn.metrics.pairwise import cosine_similarity
12
  from spacy.matcher import PhraseMatcher
@@ -25,6 +26,7 @@ from SkillExtract import SkillExtractorDetails
25
  from ExtractContentsFromFile import ExtractContentFromFile
26
  from RemoveSkills import RemoveSkill
27
  from AddSkillDetails import AddSkill
 
28
  import ClassModals
29
  import os
30
  os.environ['HF_HOME'] = '/hug/cache/'
@@ -34,6 +36,7 @@ app = FastAPI()
34
  nlp = spacy.load("en_core_web_lg")
35
  # init skill extractor
36
  skill_extractor = SkillExtractor(nlp, SKILL_DB, PhraseMatcher)
 
37
 
38
  @app.get("/")
39
  async def root():
@@ -104,5 +107,11 @@ def UploadProfile(file: bytes = File(...), FileName: str = "sample.pdf"):
104
  df = pd.DataFrame(data)
105
  return parse_csv(df)
106
 
 
 
 
 
 
 
107
  #return JSONResponse(content={"message": "Here's your interdimensional portal." , "mes1":"data2"})
108
  #https://vaibhav84-resumeapi.hf.space/docs
 
7
  import psycopg2
8
  import pandas as pd
9
  import numpy as np
10
+ from sentence_transformers import SentenceTransformer
11
  import spacy
12
  from sklearn.metrics.pairwise import cosine_similarity
13
  from spacy.matcher import PhraseMatcher
 
26
  from ExtractContentsFromFile import ExtractContentFromFile
27
  from RemoveSkills import RemoveSkill
28
  from AddSkillDetails import AddSkill
29
+ from SkillMatcher import SkillMatch
30
  import ClassModals
31
  import os
32
  os.environ['HF_HOME'] = '/hug/cache/'
 
36
  nlp = spacy.load("en_core_web_lg")
37
  # init skill extractor
38
  skill_extractor = SkillExtractor(nlp, SKILL_DB, PhraseMatcher)
39
+ model = SentenceTransformer('all-MiniLM-L6-v2')
40
 
41
  @app.get("/")
42
  async def root():
 
107
  df = pd.DataFrame(data)
108
  return parse_csv(df)
109
 
110
+ @app.get("/GetMatchProfileByJobId/")
111
+ def RemoveSkills(JobId : int):
112
+
113
+ return SkillMatch.SkillMatcher(model,db_params,JobId)
114
+
115
+
116
  #return JSONResponse(content={"message": "Here's your interdimensional portal." , "mes1":"data2"})
117
  #https://vaibhav84-resumeapi.hf.space/docs