Vaibhav84 commited on
Commit
4aba3cf
·
1 Parent(s): 2b47975
Files changed (2) hide show
  1. SkillExtract.py +1 -1
  2. app.py +2 -2
SkillExtract.py CHANGED
@@ -3,7 +3,7 @@ import psycopg2
3
  from psycopg2 import sql
4
  import pandas as pd
5
  import re
6
- class SkillExtractor:
7
 
8
  def GetSkillId(skillname,jdmasterid,db_params):
9
  #Fetching skill id from skillmaster
 
3
  from psycopg2 import sql
4
  import pandas as pd
5
  import re
6
+ class SkillExtractorDetails:
7
 
8
  def GetSkillId(skillname,jdmasterid,db_params):
9
  #Fetching skill id from skillmaster
app.py CHANGED
@@ -27,7 +27,7 @@ warnings.filterwarnings('ignore')
27
  #Custom Classes for endpoints
28
  from DbConnection import DbConnection
29
  from UploadFile import UploadOpenFile
30
- from SkillExtract import SkillExtractor
31
 
32
  import os
33
  os.environ['HF_HOME'] = '/hug/cache/'
@@ -68,4 +68,4 @@ def UploadFileDetails(file_data: FileDetails):
68
 
69
  @app.post("/ExtractSkills/")
70
  def ExtractSkills(skill_data: SkillDetails):
71
- return SkillExtractor.SkillExtract(db_params,skill_extractor,skill_data.skillid)
 
27
  #Custom Classes for endpoints
28
  from DbConnection import DbConnection
29
  from UploadFile import UploadOpenFile
30
+ from SkillExtract import SkillExtractorDetails
31
 
32
  import os
33
  os.environ['HF_HOME'] = '/hug/cache/'
 
68
 
69
  @app.post("/ExtractSkills/")
70
  def ExtractSkills(skill_data: SkillDetails):
71
+ return SkillExtractorDetails.SkillExtract(db_params,skill_extractor,skill_data.skillid)