Tonycrux commited on
Commit
f81821e
·
verified ·
1 Parent(s): 8aa1a74

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -0
app.py CHANGED
@@ -1,4 +1,5 @@
1
  import re
 
2
  import joblib
3
  from fastapi import FastAPI, Request
4
  from pydantic import BaseModel
@@ -14,6 +15,10 @@ app.add_middleware(
14
  allow_headers=["*"],
15
  )
16
 
 
 
 
 
17
  # Load model and vectorizer
18
  model = joblib.load("team_classifier_model.joblib")
19
  vectorizer = joblib.load("tfidf_vectorizer.joblib")
 
1
  import re
2
+ import os
3
  import joblib
4
  from fastapi import FastAPI, Request
5
  from pydantic import BaseModel
 
15
  allow_headers=["*"],
16
  )
17
 
18
+ os.environ["HF_HOME"] = "/tmp"
19
+ os.environ["TRANSFORMERS_CACHE"] = "/tmp"
20
+ os.environ["SENTENCE_TRANSFORMERS_HOME"] = "/tmp"
21
+
22
  # Load model and vectorizer
23
  model = joblib.load("team_classifier_model.joblib")
24
  vectorizer = joblib.load("tfidf_vectorizer.joblib")