Spaces:
Running
on
Zero
Running
on
Zero
Ultra+Supreme Remove src_lang and tgt_lang in load
Browse files
app.py
CHANGED
@@ -32,13 +32,13 @@ def translate_nano_large(text, model_path):
|
|
32 |
# Translation function for Ultra and Supreme models
|
33 |
def translate_ultra_supreme(text, model_path):
|
34 |
model = AutoModelForSeq2SeqLM.from_pretrained(model_path, token=TOKEN)
|
35 |
-
tokenizer = AutoTokenizer.from_pretrained(model_path,
|
36 |
translator = pipeline(
|
37 |
"translation",
|
38 |
model=model,
|
39 |
tokenizer=tokenizer,
|
40 |
max_length=512,
|
41 |
-
src_lang="eng_Latn",
|
42 |
tgt_lang="ary_Arab"
|
43 |
)
|
44 |
translation = translator(text)[0]['translation_text']
|
|
|
32 |
# Translation function for Ultra and Supreme models
|
33 |
def translate_ultra_supreme(text, model_path):
|
34 |
model = AutoModelForSeq2SeqLM.from_pretrained(model_path, token=TOKEN)
|
35 |
+
tokenizer = AutoTokenizer.from_pretrained(model_path, token=TOKEN) # Remove src_lang and tgt_lang
|
36 |
translator = pipeline(
|
37 |
"translation",
|
38 |
model=model,
|
39 |
tokenizer=tokenizer,
|
40 |
max_length=512,
|
41 |
+
src_lang="eng_Latn", # Keep src_lang and tgt_lang in the pipeline
|
42 |
tgt_lang="ary_Arab"
|
43 |
)
|
44 |
translation = translator(text)[0]['translation_text']
|