Update ixc_utils.py

#23
by daluggas - opened
Files changed (1) hide show
  1. ixc_utils.py +9 -3
ixc_utils.py CHANGED
@@ -9,9 +9,15 @@ import torchvision.transforms as transforms
9
  from decord import VideoReader
10
 
11
  def get_font():
12
- truetype_url = 'https://huggingface.co/internlm/internlm-xcomposer2d5-7b/resolve/main/SimHei.ttf?download=true'
13
- ff = urlopen(truetype_url)
14
- font = ImageFont.truetype(ff, size=40)
 
 
 
 
 
 
15
  return font
16
 
17
  def padding_336(b, pad=336):
 
9
  from decord import VideoReader
10
 
11
  def get_font():
12
+ current_dir = os.path.dirname(os.path.abspath(__file__))
13
+ font_path = os.path.join(current_dir, 'SimHei.ttf')
14
+
15
+ if os.path.isfile(font_filename):
16
+ font = ImageFont.truetype(font_path, size=40)
17
+ else:
18
+ truetype_url = 'https://huggingface.co/internlm/internlm-xcomposer2d5-7b/resolve/main/SimHei.ttf?download=true'
19
+ ff = urlopen(truetype_url)
20
+ font = ImageFont.truetype(ff, size=40)
21
  return font
22
 
23
  def padding_336(b, pad=336):