File size: 675 Bytes
7ab3fbc 19a9f33 7ab3fbc 19a9f33 7ab3fbc 19a9f33 7ab3fbc 19533dc 7ab3fbc 19533dc |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 |
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, multi_label=False)
return result['labels'][0]
iface = gr.Interface(
fn=classify,
inputs=[
gr.Textbox(label="Название товара"),
gr.Textbox(label="Категории (через запятую)", value="Техника, Овощи, Инструменты")
],
outputs=gr.Textbox(label="Категория")
)
iface.launch() |