Lord-Raven commited on
Commit ·
259344b
1
Parent(s): 81bb8c3
Trying to fix parameter issue.
Browse files
app.py
CHANGED
|
@@ -14,7 +14,8 @@ classifier = pipeline(
|
|
| 14 |
model=model,
|
| 15 |
tokenizer=tokenizer,
|
| 16 |
top_k=None,
|
| 17 |
-
function_to_apply="sigmoid",
|
|
|
|
| 18 |
)
|
| 19 |
|
| 20 |
def predict(param_0):
|
|
@@ -26,5 +27,4 @@ demo = gr.Interface(
|
|
| 26 |
outputs = 'text',
|
| 27 |
)
|
| 28 |
|
| 29 |
-
# demo = gr.Interface.from_pipeline(classifier)
|
| 30 |
demo.launch()
|
|
|
|
| 14 |
model=model,
|
| 15 |
tokenizer=tokenizer,
|
| 16 |
top_k=None,
|
| 17 |
+
function_to_apply="sigmoid",
|
| 18 |
+
binary_output=True,
|
| 19 |
)
|
| 20 |
|
| 21 |
def predict(param_0):
|
|
|
|
| 27 |
outputs = 'text',
|
| 28 |
)
|
| 29 |
|
|
|
|
| 30 |
demo.launch()
|