Upload mod.py

mod.py

@@ -5,9 +5,12 @@ from PIL import Image
 from pathlib import Path
 import gc
 import subprocess
-from env import num_cns, model_trigger, HF_TOKEN, CIVITAI_API_KEY, download_lora_list, directory_loras
-from modutils import download_things
 import os
+import re
+from translatepy import Translator
+from huggingface_hub import HfApi
+from env import num_cns, model_trigger, HF_TOKEN, CIVITAI_API_KEY, DOWNLOAD_LORA_LIST, DIRECTORY_LORAS
+from modutils import download_things
 
 
 subprocess.run('pip install flash-attn --no-build-isolation', env={'FLASH_ATTENTION_SKIP_CUDA_BUILD': "TRUE"}, shell=True)
@@ -19,15 +22,15 @@ control_images = [None] * num_cns
 control_modes = [-1] * num_cns
 control_scales = [0] * num_cns
 
+
 # Download stuffs
-download_lora = ", ".join(download_lora_list)
+download_lora = ", ".join(DOWNLOAD_LORA_LIST)
 for url in [url.strip() for url in download_lora.split(',')]:
     if not os.path.exists(f"./loras/{url.split('/')[-1]}"):
-        download_things(directory_loras, url, HF_TOKEN, CIVITAI_API_KEY)
+        download_things(DIRECTORY_LORAS, url, HF_TOKEN, CIVITAI_API_KEY)
 
 
 def is_repo_name(s):
-    import re
     return re.fullmatch(r'^[^/,\s\"\']+/[^/,\s\"\']+$', s)
 
 
@@ -43,7 +46,6 @@ def is_repo_exists(repo_id):
         return True # for safe
 
 
-from translatepy import Translator
 translator = Translator()
 def translate_to_en(input: str):
     try:
@@ -66,7 +68,6 @@ def clear_cache():
 
 
 def get_repo_safetensors(repo_id: str):
-    from huggingface_hub import HfApi
     api = HfApi(token=HF_TOKEN)
     try:
         if not is_repo_name(repo_id) or not is_repo_exists(repo_id): return gr.update(value="", choices=[])
@@ -364,4 +365,4 @@ load_prompt_enhancer.zerogpu = True
 fuse_loras.zerogpu = True
 preprocess_image.zerogpu = True
 get_control_params.zerogpu = True
-clear_cache.zerogpu = True
+clear_cache.zerogpu = True