| from transformers import pipeline, AutoTokenizer, AutoModelForSeq2SeqLM | |
| import gradio as gr | |
| # Завантаження токенізатора і моделі для перекладу з англійської на українську | |
| tokenizer = AutoTokenizer.from_pretrained("./model") | |
| model = AutoModelForSeq2SeqLM.from_pretrained("./model") | |
| # Створення пайплайну для перекладу тексту | |
| translation_pipeline = pipeline("translation_en_to_uk", model=model, tokenizer=tokenizer) | |
| # Функція для перекладу тексту на основі вхідного запиту | |
| def translate(prompt): | |
| translated_text = translation_pipeline(prompt)[0]["translation_text"] | |
| return translated_text | |
| # Створення інтерфейсу з використанням Gradio | |
| gr.Interface(fn=translate, inputs="text", outputs="text").launch() |