NomClass / app.py
hh1199's picture
Update app.py
37e830f verified
raw
history blame
779 Bytes
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()