Lord-Raven
commited on
Commit
·
9d79f7e
1
Parent(s):
8a0cad7
Trying to fix parameter issue.
Browse files
app.py
CHANGED
@@ -1,4 +1,5 @@
|
|
1 |
import gradio as gr
|
|
|
2 |
from transformers import AutoTokenizer, pipeline
|
3 |
from optimum.onnxruntime import ORTModelForSequenceClassification
|
4 |
|
@@ -24,7 +25,7 @@ def predict(param_0):
|
|
24 |
demo = gr.Interface(
|
25 |
fn = predict,
|
26 |
inputs = 'text',
|
27 |
-
outputs = '
|
28 |
)
|
29 |
|
30 |
demo.launch()
|
|
|
1 |
import gradio as gr
|
2 |
+
import json
|
3 |
from transformers import AutoTokenizer, pipeline
|
4 |
from optimum.onnxruntime import ORTModelForSequenceClassification
|
5 |
|
|
|
25 |
demo = gr.Interface(
|
26 |
fn = predict,
|
27 |
inputs = 'text',
|
28 |
+
outputs = 'json',
|
29 |
)
|
30 |
|
31 |
demo.launch()
|