Spaces:
Runtime error
Runtime error
kargaranamir
commited on
Commit
•
e90bc4c
1
Parent(s):
bc59854
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 |
|