Tayaba171 commited on
Commit
fca0a9d
1 Parent(s): a4c4555

Update CALTextModel.py

Browse files
Files changed (1) hide show
  1. CALTextModel.py +1 -1
CALTextModel.py CHANGED
@@ -635,7 +635,7 @@ def predict(CALTEXT, images, x_mask):
635
  xx_mask = x_mask[img_ind][tf.newaxis, ... ]
636
  anno,hidden_state_0=execute_model(xx,xx_mask,CALTEXT)
637
 
638
- sample, score,hypalpha=CALTextModel.get_sample(anno, hidden_state_0,10, 130, False, False, CALTEXT)
639
 
640
 
641
  score = score / np.array([len(s) for s in sample])
 
635
  xx_mask = x_mask[img_ind][tf.newaxis, ... ]
636
  anno,hidden_state_0=execute_model(xx,xx_mask,CALTEXT)
637
 
638
+ sample, score,hypalpha=get_sample(anno, hidden_state_0,10, 130, False, False, CALTEXT)
639
 
640
 
641
  score = score / np.array([len(s) for s in sample])