From 7620c729fa90b2ad6a2ef34ef4d9d442d7578a20 Mon Sep 17 00:00:00 2001
From: Superag <superag@gmail.com>
Date: Mon, 24 Apr 2023 16:05:49 -0700
Subject: [PATCH] update install.py

---
 install.py | 12 +++++++-----
 1 file changed, 7 insertions(+), 5 deletions(-)

diff --git a/install.py b/install.py
index e723044..58ad0d1 100644
--- a/install.py
+++ b/install.py
@@ -2,8 +2,6 @@ import launch
 import os
 import gzip
 import io
-from scripts.runtime.msai_prelude import MiaoshouPrelude
-
 
 def install_preset_models_if_needed():
     assets_folder = os.path.join(os.path.dirname(os.path.realpath(__file__)), "assets")
@@ -20,9 +18,13 @@ def install_preset_models_if_needed():
                         model_file.write(content)
 
 def create_folders():
-    prelude = MiaoshouPrelude()
-    os.mkdir(prelude.cover_folder)
-    os.mkdir(prelude.cache_folder)
+    cover_folder = os.path.join(os.path.dirname(os.path.realpath(__file__)), "cover")
+    cache_folder = os.path.join(os.path.dirname(os.path.realpath(__file__)), "cache")
+
+    if not os.path.exists(cover_folder):
+        os.mkdir('cover')
+    if not os.path.exists(cache_folder):
+        os.mkdir('cache')
 
 req_file = os.path.join(os.path.dirname(os.path.realpath(__file__)), "requirements.txt")