Update app.py

app.py

@@ -2,10 +2,10 @@ import gradio as gr
 from transformers import pipeline
 
 # Загружаем модель
-classifier = pipeline("text-classification", model="sberbank-ai/ruRoberta-large")
+classifier = pipeline("text-classification", model="sileod/rubert-tiny-for-ner")
 
 # Категории в порядке, соответствующем меткам модели
-CATEGORIES = ["Овощи", "Инструменты", "Коробки", "Прочее"]  # Измените на ваши категории!
+CATEGORIES = ["Техника", "Овощи", "Инструменты", "Коробки", "Прочее"]  # Измените на ваши категории!
 
 def classify(item: str, categories: list) -> str:
     prompt = f"""
@@ -21,8 +21,8 @@ def classify(item: str, categories: list) -> str:
 iface = gr.Interface(
     fn=classify,
     inputs=[
-        gr.Textbox(label="Название товара Сбер"), 
-        gr.Textbox(label="Категории (через запятую)", value="Овощи, Инструменты, Коробки, Прочее")
+        gr.Textbox(label="Название товара rubert-tiny-for-ner"), 
+        gr.Textbox(label="Категории (через запятую)", value="Техника, Овощи, Инструменты, Коробки, Прочее")
     ],
     outputs=gr.Textbox(label="Категория")
 )