Spaces:

Omnibus
/

HF_Space_Downloader

Running

Omnibus commited on Oct 20, 2023

Commit

9f4c9a8

1 Parent(s): acb66b7

Update dl.py

Files changed (1) hide show

dl.py CHANGED Viewed

@@ -117,15 +117,16 @@ def show_f2(repo,name,token):
 def show_f_cont(repo,name,file,token):
     html_text = '<html>\n<body>\n<div>\n'
-    if (".png" or ".jpg" or ".gif" or ".webm" or ".mp4") in file:
-        html_text += f'<object data="https://huggingface.co/spaces/{repo}/{name}/resolve/main/{file}" style="width:100%;font-size:small;"></object>'
-        out_text = "Image File"
-    else:
-        r = requests.get(f'https://huggingface.co/spaces/{repo}/{name}/raw/main/{file}')
-        text=r.text
-        html_text += f'<pre style="text-wrap:pretty;">{text}</pre>\n'
-        out_text = r.text
     html_text += '</div>\n</body>\n</html>'
     return(html_text,out_text)

 def show_f_cont(repo,name,file,token):
     html_text = '<html>\n<body>\n<div>\n'
+    images=[".png" , ".jpg" , ".gif" , ".webm" , ".mp4"]
+    for x in images:
+        if x in file:
+            html_text += f'<object data="https://huggingface.co/spaces/{repo}/{name}/resolve/main/{file}" style="width:100%;font-size:small;"></object>'
+            out_text = "Image File"
+        else:
+            r = requests.get(f'https://huggingface.co/spaces/{repo}/{name}/raw/main/{file}')
+            text=r.text
+            html_text += f'<pre style="text-wrap:pretty;">{text}</pre>\n'
+            out_text = r.text
     html_text += '</div>\n</body>\n</html>'
     return(html_text,out_text)