Spaces:
Sleeping
Sleeping
vkovacs
commited on
Commit
·
56b1f1f
1
Parent(s):
fb05782
list of tuples instead of list of dicts
Browse files
app.py
CHANGED
@@ -72,11 +72,7 @@ def predict_wrapper(text, language):
|
|
72 |
results = []
|
73 |
for sentence in sentences:
|
74 |
label, probability = predict(sentence, model_id, tokenizer_id)
|
75 |
-
results.append(
|
76 |
-
"Sentence": sentence,
|
77 |
-
"Prediction": label,
|
78 |
-
"Probability": probability
|
79 |
-
})
|
80 |
|
81 |
output_info = f'Prediction made using the <a href="https://huggingface.co/{model_id}">{model_id}</a> model.'
|
82 |
return results, output_info
|
|
|
72 |
results = []
|
73 |
for sentence in sentences:
|
74 |
label, probability = predict(sentence, model_id, tokenizer_id)
|
75 |
+
results.append([sentence, label, probability])
|
|
|
|
|
|
|
|
|
76 |
|
77 |
output_info = f'Prediction made using the <a href="https://huggingface.co/{model_id}">{model_id}</a> model.'
|
78 |
return results, output_info
|