Omnibus commited on
Commit
2f4adf3
1 Parent(s): 5f9b0b9

Update player.py

Browse files
Files changed (1) hide show
  1. player.py +6 -1
player.py CHANGED
@@ -4,6 +4,8 @@ import requests
4
  import json
5
  import os
6
  from PIL import Image
 
 
7
  token = ""
8
  source_dataset = "jbilcke-hf/ai-tube-index"
9
  api = HfApi(token=token)
@@ -140,7 +142,10 @@ def build_space(index_sel, vid_file, t_name=None, token=None):
140
  return gr.HTML(f"""{str(e)}""")
141
 
142
  if t_image != None:
143
- img1 = Image.open(t_image)
 
 
 
144
  img1.thumbnail((500,500), Image.Resampling.LANCZOS)
145
  img1.save(f'tmpim-{uid}.png')
146
  output_pro=cv2.imread(f'tmpim-{uid}.png')
 
4
  import json
5
  import os
6
  from PIL import Image
7
+ import urllib.request
8
+
9
  token = ""
10
  source_dataset = "jbilcke-hf/ai-tube-index"
11
  api = HfApi(token=token)
 
142
  return gr.HTML(f"""{str(e)}""")
143
 
144
  if t_image != None:
145
+
146
+ urllib.request.urlretrieve(f'{t_image}',"gfg.png")
147
+ img1 = Image.open("gfg.png")
148
+ #img1 = Image.open(t_image)
149
  img1.thumbnail((500,500), Image.Resampling.LANCZOS)
150
  img1.save(f'tmpim-{uid}.png')
151
  output_pro=cv2.imread(f'tmpim-{uid}.png')