MdSourav76046 commited on
Commit
bc3c30b
·
verified ·
1 Parent(s): 987fe62

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -58,9 +58,9 @@ async def load_model():
58
  # Set cache directory if not already set
59
  import os
60
  if not os.environ.get("TRANSFORMERS_CACHE"):
61
- os.environ["TRANSFORMERS_CACHE"] = "/app/cache"
62
  if not os.environ.get("HF_HOME"):
63
- os.environ["HF_HOME"] = "/app/cache"
64
 
65
  # Determine device
66
  device = "cuda" if torch.cuda.is_available() else "cpu"
 
58
  # Set cache directory if not already set
59
  import os
60
  if not os.environ.get("TRANSFORMERS_CACHE"):
61
+ os.environ["TRANSFORMERS_CACHE"] = "/tmp"
62
  if not os.environ.get("HF_HOME"):
63
+ os.environ["HF_HOME"] = "/tmp"
64
 
65
  # Determine device
66
  device = "cuda" if torch.cuda.is_available() else "cpu"