Update app.py

app.py

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