r1208 commited on
Commit
0312d8a
·
verified ·
1 Parent(s): 564a2d1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -16,7 +16,7 @@ def main():
16
  return tokens_list
17
 
18
 
19
- def translate(text, tokenizer, model, bad_words_ids)):
20
  # Prepare the prompt
21
  messages = f"Translate from Korean to English: {text}"
22
  input_ids = tokenizer.apply_chat_template(messages, tokenize=True, add_generation_prompt=True, return_tensors="pt")
 
16
  return tokens_list
17
 
18
 
19
+ def translate(text, tokenizer, model, bad_words_ids):
20
  # Prepare the prompt
21
  messages = f"Translate from Korean to English: {text}"
22
  input_ids = tokenizer.apply_chat_template(messages, tokenize=True, add_generation_prompt=True, return_tensors="pt")