davanstrien HF staff commited on
Commit
9b4982c
1 Parent(s): d92088b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -73,7 +73,7 @@ classifier = pipeline('text-classification',model=model, tokenizer=tokenizer, to
73
 
74
  def predict(text):
75
  predictions = classifier(text)
76
- return {pred['label']: pred['score'] for pred in predictions}
77
 
78
 
79
 
 
73
 
74
  def predict(text):
75
  predictions = classifier(text)
76
+ return {pred['label']: float(pred['score']) for pred in predictions}
77
 
78
 
79