Hanna Hjelmeland commited on
Commit
cfb23d9
·
1 Parent(s): 483b69e
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -66,7 +66,7 @@ def classify_text(test_text, selected_model):
66
  predicted_performance = performance_labels[predicted_class]
67
  predicted_labels.append(predicted_performance)
68
 
69
- return dict(zip(categories, map(float,performance_labels)))
70
 
71
  # Cell
72
  label = gr.outputs.Label()
 
66
  predicted_performance = performance_labels[predicted_class]
67
  predicted_labels.append(predicted_performance)
68
 
69
+ return dict(zip(categories, map(str,predicted_labels)))
70
 
71
  # Cell
72
  label = gr.outputs.Label()