gouravbhadraDev commited on
Commit
042861c
·
verified ·
1 Parent(s): 812d4f6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -0
app.py CHANGED
@@ -5,6 +5,10 @@ from bs4 import BeautifulSoup
5
  from transformers import pipeline
6
  import torch
7
  import re
 
 
 
 
8
 
9
  app = FastAPI()
10
 
 
5
  from transformers import pipeline
6
  import torch
7
  import re
8
+ import os
9
+
10
+ os.environ["HF_HOME"] = "/home/user/huggingface"
11
+ os.environ["TRANSFORMERS_CACHE"] = "/home/user/huggingface"
12
 
13
  app = FastAPI()
14