HF_Space_Downloader

Sleeping

Omnibus commited on Oct 20, 2023

Commit

141a9a8

•

1 Parent(s): 7a71d03

Update dl.py

Files changed (1) hide show

dl.py CHANGED Viewed

@@ -118,15 +118,18 @@ def show_f2(repo,name,token):
 def show_f_cont(repo,name,file,token):
-    r = requests.get(f'https://huggingface.co/spaces/{repo}/{name}/raw/main/{file}')
-    text=r.text
     html_text = '<html>\n<body>\n'
-    #for line in text.split('\n'):
-    html_text += f'<pre>{text}</pre>\n'
     html_text += '</body>\n</html>'
-    return(html_text,r.text)

 def show_f_cont(repo,name,file,token):
     html_text = '<html>\n<body>\n'
+    if ".png" or ".jpg" or ".gif" or ".webm" or ".mp4" in file:
+        html_text += f'<object data="https://huggingface.co/spaces/{repo}/{name}/raw/main/{file}"'
+        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>{text}</pre>\n'
+        out_text = r.text
     html_text += '</body>\n</html>'
+    return(html_text,out_text)