File size: 942 Bytes
7ab3fbc eca5009 86bd5a4 eca5009 8abab19 86bd5a4 7ab3fbc 19a9f33 eca5009 86bd5a4 eca5009 1515adb 7ab3fbc eca5009 86bd5a4 8abab19 86bd5a4 eca5009 7ab3fbc 8abab19 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 |
import gradio as gr
from transformers import pipeline
# Используем проверенную модель
classifier = pipeline(
"zero-shot-classification",
model="cointegrated/rubert-tiny2",
device=-1
)
def classify(item: str, categories: str) -> str:
categories_list = [c.strip() for c in categories.split(",")]
result = classifier(
item,
candidate_labels=categories_list,
multi_label=False
)
return f"{result['labels'][0]} (score: {result['scores'][0]:.2f})"
iface = gr.Interface(
fn=classify,
inputs=[
gr.Textbox(label="Товар"),
gr.Textbox(label="Категории", value="Инструменты, Овощи, Техника")
],
outputs=gr.Textbox(label="Результат"),
examples=[
["Молоток", "Инструменты, Овощи"],
["Морковь", "Овощи, Фрукты"]
]
)
iface.launch() |