fix bug

bmd1905 · Oct 11, 2021 · 8fda045 · 8fda045
1 parent 549f5ee
commit 8fda045
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/vietocr/tool/predictor.py b/vietocr/tool/predictor.py
@@ -78,7 +78,7 @@ def predict_batch(self, imgs, return_prob=False):
                 sents[j] = sent[i]
                 probs[j] = prob[i]
 
-    if return_prob: 
+        if return_prob: 
             return sents, probs
         else: 
             return sents