kargaranamir commited on
Commit
e90bc4c
1 Parent(s): bc59854

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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