Upload main_dl_file_en.py
Browse files
files_cells/python/en/main_dl_file_en.py
ADDED
|
@@ -0,0 +1,47 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
import os
|
| 2 |
+
from IPython.utils import capture
|
| 3 |
+
|
| 4 |
+
def check_colab_subscription():
|
| 5 |
+
return (os.sysconf('SC_PAGE_SIZE') * os.sysconf('SC_PHYS_PAGES') / (1024. ** 3) <= 20)
|
| 6 |
+
free_plan = check_colab_subscription()
|
| 7 |
+
|
| 8 |
+
def detect_environment():
|
| 9 |
+
environments = {
|
| 10 |
+
'COLAB_GPU': ('Google Colab', "/root" if free_plan else "/content"),
|
| 11 |
+
'KAGGLE_URL_BASE': ('Kaggle', "/kaggle/working/content"),
|
| 12 |
+
'SAGEMAKER_INTERNAL_IMAGE_URI': ('SageMaker Studio Lab', "/home/studio-lab-user/content")
|
| 13 |
+
}
|
| 14 |
+
|
| 15 |
+
for env_var, (environment, path) in environments.items():
|
| 16 |
+
if env_var in os.environ:
|
| 17 |
+
return environment, path
|
| 18 |
+
|
| 19 |
+
print("\033[31mError: an unsupported runtime environment was detected.\n\033[34mSupported environments:\033[0m Google Colab, Kaggle, Sagemaker Studio Lab")
|
| 20 |
+
return None, None
|
| 21 |
+
|
| 22 |
+
env, root_path = detect_environment()
|
| 23 |
+
if env and root_path:
|
| 24 |
+
webui_path = f"{root_path}/sdw"
|
| 25 |
+
|
| 26 |
+
print(f"Runtime environment: \033[33m{env}\033[0m")
|
| 27 |
+
# Colab Plan
|
| 28 |
+
if env == "Google Colab":
|
| 29 |
+
print(f"Colab Pro subscription: \033[34m{not free_plan}\033[0m")
|
| 30 |
+
print(f"File location: \033[32m{root_path}\033[0m")
|
| 31 |
+
|
| 32 |
+
print("Please wait for the files to download.... 👀", end='')
|
| 33 |
+
with capture.capture_output() as cap:
|
| 34 |
+
files = [f'widgets_{lang}.py', f'downloading_{lang}.py', f'launch_{lang}.py', f'auto-cleaner_{lang}.py']
|
| 35 |
+
folder_path = f'{root_path}/file_cell'
|
| 36 |
+
|
| 37 |
+
if os.path.exists(folder_path):
|
| 38 |
+
get_ipython().system('rm -rf {folder_path}')
|
| 39 |
+
|
| 40 |
+
get_ipython().system('mkdir -p {folder_path}')
|
| 41 |
+
|
| 42 |
+
for file in files:
|
| 43 |
+
get_ipython().system('wget -q https://huggingface.co/NagisaNao/fast_repo/resolve/main/files_cells/python/{lang}/{file} -O {folder_path}/{file}')
|
| 44 |
+
|
| 45 |
+
del cap
|
| 46 |
+
print("\rDone! Now you can run the cells below. ☄️" + " "*30)
|
| 47 |
+
|