Forráskód Böngészése

fix en_PP-OCR visua (#4355)

zhangyubo0722 4 hónapja
szülő
commit
5fe3c0080a

+ 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 (