Update app.py
Browse files
app.py
CHANGED
|
@@ -210,7 +210,7 @@ def compute(sentences, version = 'v3'):
|
|
| 210 |
|
| 211 |
output = model_choice.predict(sent)
|
| 212 |
|
| 213 |
-
output_label = output[0][0].split('__')[-1]
|
| 214 |
output_prob = max(min(output[1][0], 1), 0)
|
| 215 |
output_label_language = output_label.split('_')[0]
|
| 216 |
|
|
|
|
| 210 |
|
| 211 |
output = model_choice.predict(sent)
|
| 212 |
|
| 213 |
+
output_label = output[0][0].split('__')[-1].replace('_Hans', '_Hani').replace('_Hant', '_Hani')
|
| 214 |
output_prob = max(min(output[1][0], 1), 0)
|
| 215 |
output_label_language = output_label.split('_')[0]
|
| 216 |
|