Update app.py

app.py

@@ -1,12 +1,12 @@
 import gradio as gr
 from transformers import pipeline
 
-# Загружаем модели для анализа тональности, суммаризации текста, генерации подписей к изображениям, ответов на вопросы и расшифровки аудио
+# Загружаем модели для анализа тональности, суммаризации текста, генерации подписей к изображениям, ответов на вопросы и перевода текста
 sentiment_pipeline = pipeline("sentiment-analysis")
 summarization_pipeline = pipeline("summarization")
 image_captioning_pipeline = pipeline("image-to-text")
 qa_pipeline = pipeline("question-answering")
-speech_to_text_pipeline = pipeline("automatic-speech-recognition", model="openai/whisper-small")
+translation_pipeline = pipeline("translation_en_to_ru", model="Helsinki-NLP/opus-mt-en-ru")
 
 # Функция для анализа тональности текста
 def analyze_sentiment(text):
@@ -28,10 +28,10 @@ def answer_question(context, question):
     result = qa_pipeline(question=question, context=context)
     return f"Answer: {result['answer']}, Confidence: {result['score']:.4f}"
 
-# Функция для расшифровки аудио в текст
-def transcribe_audio(audio):
-    result = speech_to_text_pipeline(audio)
-    return result['text']
+# Функция для перевода текста
+def translate_text(text):
+    result = translation_pipeline(text)
+    return result[0]['translation_text']
 
 # Примеры текстов для анализа тональности
 sentiment_examples = [
@@ -63,10 +63,11 @@ qa_examples = [
     ["Artificial intelligence is transforming industries by automating tasks and providing insights from large datasets.", "How is AI transforming industries?"]
 ]
 
-# Примеры аудио для расшифровки
-audio_examples = [
-    "https://lazypy.ro/tts/assets/audio/StreamElements_Nicole_5c029bf08a515afb40579f0cd0c028bf.mp3",  # Пример 1
-    "https://lazypy.ro/tts/assets/audio/Bing_en-US-EricNeural_e2ca58109fc0206fb0d201689ca2bc8f.mp3"   # Пример 2
+# Примеры текстов для перевода
+translation_examples = [
+    "Hello, how are you?",
+    "I love machine learning and artificial intelligence.",
+    "The weather is beautiful today."
 ]
 
 # Создаем интерфейс Gradio с вкладками
@@ -114,14 +115,14 @@ with gr.Blocks() as demo:
             examples=qa_examples,
             examples_per_page=2
         )
-    with gr.Tab("Speech-to-Text Transcription"):
+    with gr.Tab("Language Translation"):
         gr.Interface(
-            fn=transcribe_audio,
-            inputs=gr.Audio(type="filepath", label="Загрузите аудиофайл"),
+            fn=translate_text,
+            inputs=gr.Textbox(lines=2, placeholder="Введите текст на английском..."),
             outputs="text",
-            title="Расшифровка аудио в текст(сломаный)",
-            description="Загрузите аудиофайл, чтобы получить его текстовую расшифровку.",
-            examples=audio_examples,
+            title="Перевод текста (английский → русский)",
+            description="Введите текст на английском, чтобы перевести его на русский.",
+            examples=translation_examples,
             examples_per_page=2
         )