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) | |
io.launch() | |