File size: 365 Bytes
299d4eb 7bdb64e 299d4eb |
1 2 3 4 5 6 7 8 9 |
from transformers import pipeline, AutoTokenizer, AutoModelForSequenceClassification
import gradio as gr
model = AutoModelForSequenceClassification.from_pretrained("NDugar/ZSD-microsoft-v2xxlmnli")
p = pipeline("zero-shot-classification", model=model, tokenizer = "NDugar/ZSD-microsoft-v2xxlmnli")
io = gr.Interface.from_pipeline(p, theme="default")
io.launch()
|