Update app.py

app.py

@@ -1,9 +1,10 @@
 import gradio as gr
 from transformers import pipeline
 
-# Загружаем модели для анализа тональности и суммаризации текста
+# Загружаем модели для анализа тональности, суммаризации текста и генерации подписей к изображениям
 sentiment_pipeline = pipeline("sentiment-analysis")
 summarization_pipeline = pipeline("summarization")
+image_captioning_pipeline = pipeline("image-to-text")
 
 # Функция для анализа тональности текста
 def analyze_sentiment(text):
@@ -15,6 +16,11 @@ def summarize_text(text):
     result = summarization_pipeline(text, max_length=50, min_length=25, do_sample=False)
     return result[0]['summary_text']
 
+# Функция для генерации подписи к изображению
+def generate_caption(image):
+    result = image_captioning_pipeline(image)
+    return result[0]['generated_text']
+
 # Примеры текстов для анализа тональности
 sentiment_examples = [
     "I love programming, it's so much fun!",
@@ -31,6 +37,13 @@ summarization_examples = [
     "Artificial intelligence is transforming industries by automating tasks and providing insights from large datasets."
 ]
 
+# Примеры изображений для генерации подписей
+image_examples = [
+    "https://a.d-cd.net/b977306s-1920.jpg",  # Пример 1
+    "https://i.pinimg.com/originals/ba/bd/6d/babd6d37eb2dd965c7f1dfb516d54094.jpg",  # Пример 2
+    "https://get.wallhere.com/photo/sea-bay-water-beach-coast-swimming-pool-resort-island-lagoon-Caribbean-vacation-estate-leisure-ocean-tropics-2560x1440-px-geographical-feature-atoll-554636.jpg"  # Пример 3
+]
+
 # Создаем интерфейс Gradio с вкладками
 with gr.Blocks() as demo:
     with gr.Tab("Sentiment Analysis"):
@@ -53,6 +66,15 @@ with gr.Blocks() as demo:
             examples=summarization_examples,
             examples_per_page=3
         )
+    with gr.Tab("Image Captioning"):
+        gr.Interface(
+            fn=generate_caption,
+            inputs=gr.Image(type="pil", label="Загрузите изображение"),
+            outputs="text",
+            title="Генерация подписи к изображению",
+            description="Загрузите изображение, чтобы сгенерировать его описание.",
+            examples=image_examples  # Добавляем примеры изображений
+        )
 
 # Запускаем интерфейс
 demo.launch()