Sfoglia il codice sorgente

fix en_PP-OCR visua (#4355)

zhangyubo0722 4 mesi fa
parent
commit
5fe3c0080a
1 ha cambiato i file con 1 aggiunte e 1 eliminazioni
  1. 1 1
      paddlex/inference/models/text_recognition/predictor.py

+ 1 - 1
paddlex/inference/models/text_recognition/predictor.py

@@ -111,7 +111,7 @@ class TextRecPredictor(BasePredictor):
         return None, None
 
     def get_vis_font(self):
-        if self.model_name.startswith("PP-OCR"):
+        if self.model_name.startswith(("PP-OCR", "en_PP-OCR")):
             return SIMFANG_FONT
 
         if self.model_name in (