Hugging Face's logo

Spaces:
radinhas
/
hf-llm-api
Running

App Files Community
radinhas commited on
Commit
dfc1d78
·
1 Parent(s): 02e2c20

Update apis/chat_api.py

Browse files
Files changed (1) hide show
  1. apis/chat_api.py +1 -1
apis/chat_api.py CHANGED
@@ -107,7 +107,7 @@ class ChatAPIApp:
107
  time_start = time.time()
108
  #TRANSFORMERS_CACHE
109
  pretrained_model = "facebook/m2m100_1.2B"
110
- cache_dir = "apis/models/"
111
  tokenizer = M2M100Tokenizer.from_pretrained(pretrained_model, cache_dir=cache_dir)
112
  model = M2M100ForConditionalGeneration.from_pretrained(
113
  pretrained_model, cache_dir=cache_dir
 
107
  time_start = time.time()
108
  #TRANSFORMERS_CACHE
109
  pretrained_model = "facebook/m2m100_1.2B"
110
+ cache_dir = "/home/models/"
111
  tokenizer = M2M100Tokenizer.from_pretrained(pretrained_model, cache_dir=cache_dir)
112
  model = M2M100ForConditionalGeneration.from_pretrained(
113
  pretrained_model, cache_dir=cache_dir