Spaces:
Sleeping
Sleeping
vkovacs commited on
Commit ·
dd02147
1
Parent(s): cec858f
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
|