Commit
·
80aea0d
1
Parent(s):
be1b3ee
updating labels striping
Browse files
app.py
CHANGED
|
@@ -10,7 +10,8 @@ classifier = pipeline("zero-shot-classification",
|
|
| 10 |
|
| 11 |
#define a function to process your input and output
|
| 12 |
def zero_shot(doc, candidates):
|
| 13 |
-
given_labels = candidates.split(",
|
|
|
|
| 14 |
dictionary = classifier(doc, given_labels)
|
| 15 |
labels = dictionary['labels']
|
| 16 |
scores = dictionary['scores']
|
|
@@ -54,7 +55,7 @@ transactions_and_tags = [
|
|
| 54 |
gui = gr.Interface(title=title,
|
| 55 |
description=description,
|
| 56 |
fn=zero_shot,
|
| 57 |
-
inputs=[
|
| 58 |
outputs=[output],
|
| 59 |
examples=transactions_and_tags)
|
| 60 |
|
|
|
|
| 10 |
|
| 11 |
#define a function to process your input and output
|
| 12 |
def zero_shot(doc, candidates):
|
| 13 |
+
given_labels = candidates.split(",")
|
| 14 |
+
given_labels = list(map(strip,given_labels))
|
| 15 |
dictionary = classifier(doc, given_labels)
|
| 16 |
labels = dictionary['labels']
|
| 17 |
scores = dictionary['scores']
|
|
|
|
| 55 |
gui = gr.Interface(title=title,
|
| 56 |
description=description,
|
| 57 |
fn=zero_shot,
|
| 58 |
+
inputs=[input1, input2],
|
| 59 |
outputs=[output],
|
| 60 |
examples=transactions_and_tags)
|
| 61 |
|