Update app.py
Browse files
app.py
CHANGED
@@ -1,6 +1,6 @@
|
|
1 |
from transformers import pipeline
|
2 |
import gradio as gr
|
3 |
-
classifier = pipeline("zero-shot-classification", model="
|
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)
|