Vaibhav84 commited on
Commit
a3cdaed
·
1 Parent(s): 48a1d23
Files changed (1) hide show
  1. app.py +3 -4
app.py CHANGED
@@ -33,15 +33,14 @@ from DbConnection import DbConnection
33
  from UploadFile import UploadOpenFile
34
  from SkillExtract import SkillExtractorDetails
35
  import os
36
- from typing import Optional
37
  os.environ['HF_HOME'] = '/hug/cache/'
38
 
39
  app = FastAPI()
40
  class FileDetails(BaseModel):
41
  filecontents: str
42
  filename: str
43
- fileid: Optional[int]
44
- message: Optional[str]
45
 
46
 
47
  class SkillDetails(BaseModel):
@@ -126,4 +125,4 @@ def process_pdf_file(file: bytes = File(...)):
126
  text += page.extract_text()
127
  print(text)
128
  # Process saved file
129
- return {"content": text}
 
33
  from UploadFile import UploadOpenFile
34
  from SkillExtract import SkillExtractorDetails
35
  import os
 
36
  os.environ['HF_HOME'] = '/hug/cache/'
37
 
38
  app = FastAPI()
39
  class FileDetails(BaseModel):
40
  filecontents: str
41
  filename: str
42
+ fileid: str
43
+ message: str
44
 
45
 
46
  class SkillDetails(BaseModel):
 
125
  text += page.extract_text()
126
  print(text)
127
  # Process saved file
128
+ return {"content": file.filename}