Vaibhav84 commited on
Commit
bcc36a0
·
1 Parent(s): 548eb12
Files changed (1) hide show
  1. app.py +3 -0
app.py CHANGED
@@ -29,6 +29,9 @@ from DbConnection import DbConnection
29
  from UploadFile import UploadOpenFile
30
  from SkillExtract import SkillExtractor
31
 
 
 
 
32
  app = FastAPI()
33
 
34
  class FileDetails(BaseModel):
 
29
  from UploadFile import UploadOpenFile
30
  from SkillExtract import SkillExtractor
31
 
32
+ import os
33
+ os.environ['TRANSFORMERS_CACHE'] = '/hug/cache/'
34
+
35
  app = FastAPI()
36
 
37
  class FileDetails(BaseModel):