Hugging Face's logo

Spaces:
DHEIVER
/
hf-llm-api-pt
Running

App Files Community
radinhas commited on
Commit
d5c01ef
•
1 Parent(s): 643b6aa

Update apis/chat_api.py

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