Vaibhav84 commited on
Commit
4696f89
·
1 Parent(s): 5a8cf14
Files changed (1) hide show
  1. app.py +18 -2
app.py CHANGED
@@ -30,12 +30,28 @@ warnings.filterwarnings('ignore')
30
  from DbConnection import DbConnection
31
  from UploadFile import UploadOpenFile
32
  from SkillExtract import SkillExtractorDetails
33
- from APIModalsDetails import FileDetails
34
- from APIModalsDetails import SkillDetails
35
  import os
36
  os.environ['HF_HOME'] = '/hug/cache/'
37
 
38
  app = FastAPI()
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
39
 
40
  nlp = spacy.load("en_core_web_lg")
41
  # init skill extractor
 
30
  from DbConnection import DbConnection
31
  from UploadFile import UploadOpenFile
32
  from SkillExtract import SkillExtractorDetails
 
 
33
  import os
34
  os.environ['HF_HOME'] = '/hug/cache/'
35
 
36
  app = FastAPI()
37
+ class FileDetails(BaseModel):
38
+ filecontents: str
39
+ filename: str
40
+ fileid: str
41
+ message: str
42
+
43
+
44
+ class SkillDetails(BaseModel):
45
+ skillid: int
46
+ requiredSkills: str
47
+ softSkills: str
48
+ goodToHaveSkills: str
49
+
50
+ class FileResponse(BaseModel):
51
+ fileid: int
52
+ message: str
53
+
54
+
55
 
56
  nlp = spacy.load("en_core_web_lg")
57
  # init skill extractor