Rooni commited on
Commit
8c42e17
·
1 Parent(s): 5309241

Create app.py

Browse files
Files changed (1) hide show
  1. app.py +38 -0
app.py ADDED
@@ -0,0 +1,38 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import gradio as gr
2
+ import requests
3
+ import os
4
+
5
+ # Получаем токен из переменной окружения
6
+ HF_TOKEN = os.getenv("HF_TOKEN")
7
+
8
+ # Функция для запроса к Hugging Face Inference API
9
+ def translate(text, target_language):
10
+ headers = {
11
+ "Authorization": f"Bearer {HF_TOKEN}"
12
+ }
13
+ data = {
14
+ "inputs": text,
15
+ "parameters": {"target_language": target_language},
16
+ }
17
+ response = requests.post("https://api-inference.huggingface.co/models/facebook/mbart-large-50-many-to-many-mmt", headers=headers, json=data)
18
+ return response.json()[0]["translation_text"]
19
+
20
+ # Список доступных языков для перевода
21
+ languages = [
22
+ "ar_AR", "cs_CZ", "de_DE", "en_XX", "es_XX", "et_EE", "fi_FI",
23
+ "fr_XX", "gu_IN", "hi_IN", "it_IT", "ja_XX", "ko_KR", "mr_IN",
24
+ "nl_XX", "ro_RO", "ru_RU", "sv_SE", "tr_TR", "zh_CN"
25
+ ]
26
+
27
+ with gr.Blocks() as demo:
28
+ with gr.Row():
29
+ text_input = gr.Textbox(lines=3, placeholder="Введите текст для перевода")
30
+ language_dropdown = gr.Dropdown(choices=languages, value="en_XX", label="Выберите язык перевода")
31
+ with gr.Row():
32
+ translate_button = gr.Button("Перевести")
33
+ with gr.Row():
34
+ output = gr.Textbox(label="Перевод")
35
+
36
+ translate_button.click(fn=translate, inputs=[text_input, language_dropdown], outputs=output)
37
+
38
+ demo.launch()