import gradio as gr
from transformers import pipeline

# Инициализация модели
classifier = pipeline(
    "zero-shot-classification",
    model="cointegrated/rubert-tiny2"
)

def classify(item: str, categories: str) -> str:
    # Основная логика классификации
    categories_list = [c.strip() for c in categories.split(",")]
    result = classifier(item, categories_list)
    return result['labels'][0]

# Интерфейс с минимальными элементами
iface = gr.Interface(
    fn=classify,
    inputs=[
        gr.Textbox(label="Название товара"), 
        gr.Textbox(label="Категории через запятую")
    ],
    outputs=gr.Textbox(label="Результат")
)

iface.launch()