HMPhuoc commited on
Commit
518d46d
1 Parent(s): 4707c41

default value for bert prediction

Browse files
Files changed (1) hide show
  1. app.py +4 -1
app.py CHANGED
@@ -111,7 +111,10 @@ def judgePlus(x):
111
 
112
  lstm_pred = LSTM_predict(x)
113
  gru_pred = GRU_predict(x)
114
- bert_pred = BERT_predict(x)
 
 
 
115
  #bert_pred = BERT_predict(x)
116
  #print(result)
117
 
 
111
 
112
  lstm_pred = LSTM_predict(x)
113
  gru_pred = GRU_predict(x)
114
+ try:
115
+ bert_pred = BERT_predict(x)
116
+ except:
117
+ bert_pred = [0.5, 0.5, 0.5, 0.5, 0.5, 0.5]
118
  #bert_pred = BERT_predict(x)
119
  #print(result)
120