Update app.py

app.py

@@ -9,16 +9,6 @@ from pydub import AudioSegment
 # Инициализация клиента для модели Felguk/Felguk-v0
 client = InferenceClient("Felguk/Felguk-v0")
 
-# Список доступных моделей
-MODELS = {
-    "felguk v0": "Felguk/Felguk-v0",
-    "chatgpt 4.0": "gpt-4",  # Пример, замените на реальный API
-    "claude 3.5": "claude-3.5",  # Пример, замените на реальный API
-    "qwen 2.5": "qwen-2.5",  # Пример, замените на реальный API
-    "deepseek": "deepseek",  # Пример, замените на реальный API
-    "le chat": "le-chat",  # Пример, замените на реальный API
-}
-
 # Функция для обработки PDF
 def process_pdf(file):
     pdf_reader = PyPDF2.PdfReader(file)
@@ -54,13 +44,7 @@ def respond(
     temperature,
     top_p,
     file=None,
-    model_name="felguk v0",
 ):
-    # Выбор модели
-    if model_name not in MODELS:
-        yield "Ошибка: модель не найдена."
-        return
-
     # Если загружен файл, обрабатываем его
     if file is not None:
         file_type = file.name.split(".")[-1].lower()
@@ -76,21 +60,16 @@ def respond(
             file_info = "Неизвестный тип файла"
         message += f"\n[Пользователь загрузил файл: {file.name}]\n{file_info}"
 
-    # Генерация ответа в зависимости от модели
-    if model_name == "felguk v0":
-        # Используем текстовое завершение для модели Felguk/Felguk-v0
-        response = client.text_generation(
-            prompt=message,
-            max_new_tokens=max_tokens,
-            temperature=temperature,
-            top_p=top_p,
-            stream=True,
-        )
-        for token in response:
-            yield token
-    else:
-        # Здесь можно добавить логику для других моделей
-        yield f"Модель {model_name} пока не поддерживается."
+    # Генерация ответа с использованием модели Felguk/Felguk-v0
+    response = client.text_generation(
+        prompt=message,
+        max_new_tokens=max_tokens,
+        temperature=temperature,
+        top_p=top_p,
+        stream=True,
+    )
+    for token in response:
+        yield token
 
 # Функция для сброса истории чата
 def reset_chat():
@@ -105,14 +84,6 @@ with gr.Blocks() as demo:
     with gr.Row():
         new_chat_button = gr.Button("Новый чат", variant="secondary")
 
-    # Выбор модели
-    with gr.Row():
-        model_dropdown = gr.Dropdown(
-            choices=list(MODELS.keys()),
-            value="felguk v0",
-            label="Выберите модель",
-        )
-
     # Интерфейс чата
     chat_interface = gr.ChatInterface(
         respond,
@@ -128,7 +99,6 @@ with gr.Blocks() as demo:
                 label="Top-p (nucleus sampling)",
             ),
             gr.File(label="Загрузите файл (опционально)"),  # Поле для загрузки файлов
-            model_dropdown,  # Выбор модели
         ],
     )