Spaces:
Sleeping
Sleeping
vickeee465
commited on
Commit
·
8383fbb
1
Parent(s):
b26c799
sentence fix 3
Browse files
app.py
CHANGED
@@ -80,7 +80,8 @@ def predict_wrapper(text, language):
|
|
80 |
results = []
|
81 |
for sentence in sentences:
|
82 |
probs = predict(sentence, model_id, tokenizer_id)
|
83 |
-
|
|
|
84 |
|
85 |
output_info = f'Prediction was made using the <a href="https://huggingface.co/{model_id}">{model_id}</a> model.'
|
86 |
return results, output_info
|
|
|
80 |
results = []
|
81 |
for sentence in sentences:
|
82 |
probs = predict(sentence, model_id, tokenizer_id)
|
83 |
+
label, probability = get_most_probable_label(probs)
|
84 |
+
results.append([sentence, label, probability])
|
85 |
|
86 |
output_info = f'Prediction was made using the <a href="https://huggingface.co/{model_id}">{model_id}</a> model.'
|
87 |
return results, output_info
|