Lord-Raven commited on
Commit
1b99eaa
·
1 Parent(s): 01b22b1

Trying to fix parameter issue.

Browse files
Files changed (1) hide show
  1. app.py +5 -1
app.py CHANGED
@@ -20,7 +20,11 @@ classifier = pipeline(
20
 
21
  def predict(param_0):
22
  results = classifier(param_0)
23
- return results[0]
 
 
 
 
24
 
25
  demo = gr.Interface(
26
  fn = predict,
 
20
 
21
  def predict(param_0):
22
  results = classifier(param_0)
23
+ response = {
24
+ "label": results[0][0].label,
25
+ "confidences": results[0]
26
+ }
27
+ return response
28
 
29
  demo = gr.Interface(
30
  fn = predict,