|
|
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 classifier("Молоток") |
|
|
|
|
|
iface = gr.Interface( |
|
|
fn=classify, |
|
|
inputs=[ |
|
|
gr.Textbox(label="Название товара vjkjnjr"), |
|
|
gr.Textbox(label="Категории (через запятую)", value="Техника, Овощи, Инструменты, Морковка") |
|
|
], |
|
|
outputs=gr.Textbox(label="Категория") |
|
|
) |
|
|
|
|
|
iface.launch() |
