Spaces:
Runtime error
Runtime error
Update CALTextModel.py
Browse files- 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=
|
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])
|