Lord-Raven commited on
Commit
8a0cad7
·
1 Parent(s): 259344b

Trying to fix parameter issue.

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -1,5 +1,4 @@
1
  import gradio as gr
2
- import json
3
  from transformers import AutoTokenizer, pipeline
4
  from optimum.onnxruntime import ORTModelForSequenceClassification
5
 
@@ -19,7 +18,8 @@ classifier = pipeline(
19
  )
20
 
21
  def predict(param_0):
22
- return classifier(param_0)
 
23
 
24
  demo = gr.Interface(
25
  fn = predict,
 
1
  import gradio as gr
 
2
  from transformers import AutoTokenizer, pipeline
3
  from optimum.onnxruntime import ORTModelForSequenceClassification
4
 
 
18
  )
19
 
20
  def predict(param_0):
21
+ results = classifier(param_0)
22
+ return results
23
 
24
  demo = gr.Interface(
25
  fn = predict,