Hugging Face's logo

Spaces:
werent4
/
mT5TranslatorLT
Sleeping

App Files Community
Dmytro Vodianytskyi commited on
Commit
46318b8
•
1 Parent(s): 6d8648d

space updated

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -7,7 +7,7 @@ TOKENIZER = T5Tokenizer.from_pretrained('werent4/mt5TranslatorLT')
7
  MODEL = MT5ForConditionalGeneration.from_pretrained("werent4/mt5TranslatorLT")
8
  MODEL.to(DEVICE)
9
 
10
- def translate(text, max_length, num_beams, translation_way):
11
  translations_ways = {
12
  "en-lt": "<EN2LT>",
13
  "lt-en": "<LT2EN>"
 
7
  MODEL = MT5ForConditionalGeneration.from_pretrained("werent4/mt5TranslatorLT")
8
  MODEL.to(DEVICE)
9
 
10
+ def translate(text, translation_way, max_length, num_beams):
11
  translations_ways = {
12
  "en-lt": "<EN2LT>",
13
  "lt-en": "<LT2EN>"