micole66 commited on
Commit
db64b8e
1 Parent(s): b0c251d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -1,6 +1,6 @@
1
  from transformers import pipeline
2
  import gradio as gr
3
- classifier = pipeline("zero-shot-classification", model="MoritzLaurer/DeBERTa-v3-large-mnli-fever-anli-ling-wanli")
4
  def wrap_classifier(text, labels, template):
5
  labels = labels.split(",")
6
  outputs = classifier(text, labels, hypothesis_template=template)
 
1
  from transformers import pipeline
2
  import gradio as gr
3
+ classifier = pipeline("zero-shot-classification", model="oigele/Fb_improved_zeroshot")
4
  def wrap_classifier(text, labels, template):
5
  labels = labels.split(",")
6
  outputs = classifier(text, labels, hypothesis_template=template)