Update app.py

app.py

@@ -11,6 +11,7 @@ import torch
 import torchvision
 from torchvision.models.detection import fasterrcnn_resnet50_fpn
 from torchvision.transforms import functional as F
+import io
 
 # Инициализация клиента для модели HuggingFaceH4/zephyr-7b-beta
 client = InferenceClient("HuggingFaceH4/zephyr-7b-beta")
@@ -75,6 +76,16 @@ def count_objects(image):
     num_objects = len(predictions[0]['labels'])
     return f"Количество объектов на изображении: {num_objects}"
 
+# Функция для конвертации изображения
+def convert_image(image, target_format):
+    if image is None:
+        return None
+    img = Image.open(image.name)
+    buffer = io.BytesIO()
+    img.save(buffer, format=target_format)
+    buffer.seek(0)
+    return buffer
+
 # Функция для определения языка текста
 def detect_language(text):
     try:
@@ -196,6 +207,19 @@ with gr.Blocks() as demo:
             numage_button = gr.Button("Определить количество объектов")
             numage_button.click(fn=count_objects, inputs=numage_input, outputs=numage_output)
 
+        # Вкладка ConverjerIMG
+        with gr.Tab("ConverjerIMG"):
+            gr.Markdown("## ConverjerIMG: Конвертер изображений")
+            img_input = gr.File(label="Загрузите изображение", file_types=["image"])
+            img_format = gr.Dropdown(
+                choices=["JPEG", "PNG", "BMP", "GIF", "TIFF"],
+                label="Выберите формат для конвертации",
+                value="JPEG"
+            )
+            img_output = gr.File(label="Результат конвертации")
+            convert_button = gr.Button("Конвертировать")
+            convert_button.click(fn=convert_image, inputs=[img_input, img_format], outputs=img_output)
+
     # Интерфейс чата
     chat_interface = gr.ChatInterface(
         respond,