File size: 779 Bytes
7ab3fbc 37e830f 86bd5a4 37e830f 86bd5a4 7ab3fbc 19a9f33 37e830f eca5009 37e830f 1515adb 37e830f 7ab3fbc 37e830f 86bd5a4 37e830f 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 |
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() |