HMPhuoc commited on
Commit
13b8bac
·
1 Parent(s): 3679935
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -141,7 +141,7 @@ def judgeBert(x):
141
  try:
142
  bert_pred = BERT_predict(x)
143
  except:
144
- bert_pred = [0,0,0,0,0,0]
145
 
146
 
147
  return_result = 'Result'
@@ -150,6 +150,6 @@ def judgeBert(x):
150
  #result_bert = np.round(bert_pred, 2)
151
 
152
  for i in range(6):
153
- result.append(result_bert[i])
154
 
155
  return (result)
 
141
  try:
142
  bert_pred = BERT_predict(x)
143
  except:
144
+ bert_pred = np.zeros(6, dtype=float)
145
 
146
 
147
  return_result = 'Result'
 
150
  #result_bert = np.round(bert_pred, 2)
151
 
152
  for i in range(6):
153
+ result.append((result_bert[i])/1)
154
 
155
  return (result)