Update dl.py

dl.py

@@ -70,11 +70,15 @@ def show_f(repo,name,token):
         img_tog=False
         for img in images:
             if img in f_name:
+                url = f"https://huggingface.co/spaces/{repo}/{name}/resolve/main/{d_app}"
+                r = requests.get(url)
+                with open(f'{name}/{f_name}', "wb") as f:
+                    f.write(r.content)                
                 #html_text += f'<object data="https://huggingface.co/spaces/{repo}/{name}/raw/main/{d_app}"'
                 #out_text = "Image File"
                 #https://huggingface.co/spaces/Omnibus/idefics_playground/resolve/main/example_images/cat_sketch.png            
-                sav_im = Image.open(f"https://huggingface.co/spaces/{repo}/{name}/resolve/main/{d_app}")
-                sav_im.save(f'{name}/{f_name}')
+                #sav_im = Image.open(f"https://huggingface.co/spaces/{repo}/{name}/resolve/main/{d_app}")
+                #sav_im.save(f'{name}/{f_name}')
                 file_list.append(Path(f'{name}/{f_name}'))
                 #file_out.append(f"https://huggingface.co/spaces/{repo}/{name}/raw/main/{d_app}")
                 img_tog=True
@@ -85,10 +89,13 @@ def show_f(repo,name,token):
             print(d_app)
             #print (r.text)
             uid = uuid.uuid4()
-  
-            file = open(f'{name}/{f_name}','w')
-            file.writelines(r.text)
+            file = open(f'{name}/{f_name}','wb')
+            file.write(r.content)
             file.close()
+  
+            #file = open(f'{name}/{f_name}','w')
+            #file.writelines(r.text)
+            #file.close()
             file_list.append(Path(f'{name}/{f_name}'))
             #file_out.append(d_app)