update

README.md

@@ -16,6 +16,6 @@ preload_from_hub:
   - openai/clip-vit-large-patch14
   - scepter-studio/scepter
   - scepter-studio/scepter_scedit
-  - scepter-studio/LAR-Gen
+  - scepter-studio/LARGEN
   - scepter-studio/stylebooth
 ---

app.py

@@ -5,14 +5,15 @@ import os
 # print(f"pip3 uninstall gradio")
 # results = os.popen(f"pip3 uninstall gradio -y")
 # print(results.readlines())
-# results = os.popen(f"ls -lR  ~/.cache/huggingface/hub")
-# print("step0:", results.readlines())
+results = os.popen(f"ls ~/.cache/huggingface/hub")
+print("step0:", results.readlines())
 results = os.popen(f"pip3 install -r requirements.txt -i https://pypi.org/simple")
 print("step1:", results.readlines())
 results = os.popen(f"pip3 uninstall scepter -y")
 print("step2:", results.readlines())
-results = os.popen(f"pip3 install scepter -i https://pypi.org/simple")
-print("step3:", results.readlines())
+scepter_whl = os.environ.get("SCEPTER_WHL","scepter")
+results = os.popen(f"pip3 install {scepter_whl} -i https://pypi.org/simple")
+print("step3:", scepter_whl, results.readlines())
 results = os.popen(f"pip3 list")
 print("step4:", results.readlines())
 #results = os.popen(f"sudo chmod 777 /data")
@@ -48,6 +49,7 @@ def link_model():
         hf_path = os.path.join(hf_root_dir, hf_name)
         cmd = f"mkdir -p {ms_path} && ln -s {hf_path}/* {ms_path}"
         os.system(cmd)
+        print(cmd)
 
 link_model()
 
@@ -69,11 +71,11 @@ class TabManager():
 config_file = os.path.join(os.path.dirname(scepter.dirname), "scepter/methods/studio/scepter_ui.yaml")
 config = Config(load=True, cfg_file=config_file)
 
-for v in config.FILE_SYSTEM:
-    # v.TEMP_DIR = os.path.join("/home", v.TEMP_DIR)
+# for v in config.FILE_SYSTEM:
+    # v.TEMP_DIR = os.path.join("./", v.TEMP_DIR)
     # v.TEMP_DIR = os.path.join("/data/.huggingface", v.TEMP_DIR)
-    v.TEMP_DIR = os.path.join("~/.cache/huggingface/hub", v.TEMP_DIR)
-    os.makedirs(v.TEMP_DIR, exist_ok=True)
+    # v.TEMP_DIR = os.path.join("~/.cache/huggingface/hub", v.TEMP_DIR)
+    # os.makedirs(v.TEMP_DIR, exist_ok=True)
 
 prepare(config)