File size: 397 Bytes
4f60f08
 
 
 
 
4cd78af
de16dcd
1
2
3
4
5
6
7
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="dark", title = "Beit Rabban Final Project")
io.launch()