Update app.py

app.py

@@ -1,45 +1,28 @@
 import gradio as gr
 from transformers import pipeline
-import time
 
-# Загружаем модель ПЕРВОЙ
-try:
-    classifier = pipeline(
-        "text-classification", 
-        model="cointegrated/rubert-tiny2",
-        truncation=True
-    )
-except Exception as e:
-    print(f"Ошибка загрузки модели: {e}")
-    raise
+# Загружаем модель
+classifier = pipeline("text-classification", model="cointegrated/rubert-tiny2")
 
-# Функция классификации
-def classify(item: str, categories: str) -> str:
-    try:
-        prompt = f"Отнеси '{item}' к категории: {categories}. Ответь только категорией."
-        result = classifier(prompt)
-        return result[0]['label']
-    except Exception as e:
-        return f"Ошибка: {str(e)}"
+# Категории в порядке, соответствующем меткам модели
+CATEGORIES = ["Овощи", "Инструменты", "Фрукты"]  # Измените на ваши категории!
 
-# Создаем интерфейс ПОСЛЕ загрузки модели
+def classify(item: str, categories: list) -> str:
+    prompt = f"Отнеси '{item}' к категории: {', '.join(categories)}. Ответь только категорией."
+    result = classifier(prompt, truncation=True)
+    
+    # Преобразуем метку (например, LABEL_0 → 0) в название категории
+    label_index = int(result[0]['label'].split("_")[1])
+    return CATEGORIES[label_index]
+
+# Интерфейс Gradio
 iface = gr.Interface(
     fn=classify,
     inputs=[
-        gr.Textbox(label="Товар"), 
+        gr.Textbox(label="Название товара"), 
         gr.Textbox(label="Категории (через запятую)", value="Овощи, Инструменты")
     ],
-    outputs="text",
-    title="Классификатор"
-)
-
-# Явно запускаем сервер
-iface.launch(
-    server_name="0.0.0.0",
-    server_port=7860,
-    show_error=True
+    outputs=gr.Textbox(label="Категория")
 )
 
-# Бесконечный цикл, чтобы сервер не завершался
-while True:
-    time.sleep(60)
+iface.launch()