gibil commited on
Commit
4f60f08
·
1 Parent(s): fe9d2f1

Create app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -0
app.py ADDED
@@ -0,0 +1,7 @@
 
 
 
 
 
 
 
 
1
+ from transformers import pipeline, AutoTokenizer, AutoModelForSequenceClassification
2
+ import gradio as gr
3
+
4
+ model = AutoModelForSequenceClassification.from_pretrained("NDugar/ZSD-microsoft-v2xxlmnli")
5
+ p = pipeline("zero-shot-classification", model=model, tokenizer = "NDugar/ZSD-microsoft-v2xxlmnli")
6
+ io = gr.Interface.from_pipeline(p, theme="default")
7
+ io.launch()