Uliana01 commited on
Commit
a52d6e5
·
verified ·
1 Parent(s): 4c0e5bc

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -1,9 +1,9 @@
1
- from transformers import pipeline, AutoTokenizer, AutoModelForSeq2SeqLM
2
  import gradio as gr
3
 
4
  # Завантаження токенізатора і моделі для перекладу з англійської на українську
5
- tokenizer = AutoTokenizer.from_pretrained("./model")
6
- model = AutoModelForSeq2SeqLM.from_pretrained("./model")
7
 
8
  # Створення пайплайну для перекладу тексту
9
  translation_pipeline = pipeline("translation_en_to_uk", model=model, tokenizer=tokenizer)
 
1
+ from transformers import pipeline, MarianMTModel, MarianTokenizer
2
  import gradio as gr
3
 
4
  # Завантаження токенізатора і моделі для перекладу з англійської на українську
5
+ tokenizer = MarianTokenizer.from_pretrained("Helsinki-NLP/opus-mt-en-uk")
6
+ model = MarianMTModel.from_pretrained("Helsinki-NLP/opus-mt-en-uk")
7
 
8
  # Створення пайплайну для перекладу тексту
9
  translation_pipeline = pipeline("translation_en_to_uk", model=model, tokenizer=tokenizer)