Lord-Raven commited on
Commit
9d79f7e
·
1 Parent(s): 8a0cad7

Trying to fix parameter issue.

Browse files
Files changed (1) hide show
  1. app.py +2 -1
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 = 'text',
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()