Kush26 commited on
Commit
cf12f14
·
verified ·
1 Parent(s): e181061

Update app/main.py

Browse files
Files changed (1) hide show
  1. app/main.py +1 -1
app/main.py CHANGED
@@ -25,7 +25,7 @@ def load_assets():
25
 
26
  global model, tokenizer, device
27
 
28
- local_cache_dir = "./hf_cache"
29
  model_file = hf_hub_download(repo_id="Kush26/Transformer_Translation", filename="model.pth", cache_dir=local_cache_dir)
30
  tokenizer_file = hf_hub_download(repo_id="Kush26/Transformer_Translation", filename="hindi-english_bpe_tokenizer.json", cache_dir=local_cache_dir)
31
 
 
25
 
26
  global model, tokenizer, device
27
 
28
+ local_cache_dir = "/tmp/hf_cache"
29
  model_file = hf_hub_download(repo_id="Kush26/Transformer_Translation", filename="model.pth", cache_dir=local_cache_dir)
30
  tokenizer_file = hf_hub_download(repo_id="Kush26/Transformer_Translation", filename="hindi-english_bpe_tokenizer.json", cache_dir=local_cache_dir)
31