Update app.py

app.py

@@ -1,33 +1,26 @@
 import gradio as gr
 from transformers import pipeline
 
-# Используем проверенную модель
+# Инициализация модели
 classifier = pipeline(
     "zero-shot-classification",
-    model="cointegrated/rubert-tiny2",
-    device=-1
+    model="cointegrated/rubert-tiny2"
 )
 
 def classify(item: str, categories: str) -> str:
+    # Основная логика классификации
     categories_list = [c.strip() for c in categories.split(",")]
-    result = classifier(
-        item,
-        candidate_labels=categories_list,
-        multi_label=False
-    )
-    return f"{result['labels'][0]} (score: {result['scores'][0]:.2f})"
+    result = classifier(item, categories_list)
+    return result['labels'][0]
 
+# Интерфейс с минимальными элементами
 iface = gr.Interface(
     fn=classify,
     inputs=[
-        gr.Textbox(label="Товар"), 
-        gr.Textbox(label="Категории", value="Инструменты, Овощи, Техника")
+        gr.Textbox(label="Название товара"), 
+        gr.Textbox(label="Категории через запятую")
     ],
-    outputs=gr.Textbox(label="Результат"),
-    examples=[
-        ["Молоток", "Инструменты, Овощи"],
-        ["Морковь", "Овощи, Фрукты"]
-    ]
+    outputs=gr.Textbox(label="Результат")
 )
 
 iface.launch()