radinhas commited on
Commit
4242d17
1 Parent(s): 1385800

Update apis/chat_api.py

Browse files
Files changed (1) hide show
  1. apis/chat_api.py +2 -1
apis/chat_api.py CHANGED
@@ -177,7 +177,8 @@ class ChatAPIApp:
177
  real_name = MODEL_MAP[target_model]
178
  read_model = AutoModelForSeq2SeqLM.from_pretrained(real_name)
179
  tokenizer = AutoTokenizer.from_pretrained(real_name)
180
- translator = pipeline("translation", model=read_model, tokenizer=tokenizer, src_lang=item.from_language, tgt_lang=item.to_language)
 
181
  result = translator(item.input_text)
182
 
183
  item_response = {
 
177
  real_name = MODEL_MAP[target_model]
178
  read_model = AutoModelForSeq2SeqLM.from_pretrained(real_name)
179
  tokenizer = AutoTokenizer.from_pretrained(real_name)
180
+ #translator = pipeline("translation", model=read_model, tokenizer=tokenizer, src_lang=item.from_language, tgt_lang=item.to_language)
181
+ translator = pipeline("translation", model=real_name, src_lang=item.from_language, tgt_lang=item.to_language)
182
  result = translator(item.input_text)
183
 
184
  item_response = {