update install.py

pull/12/head
Superag 2023-04-24 16:05:49 -07:00
parent 2148ef6c56
commit 7620c729fa
1 changed files with 7 additions and 5 deletions

View File

@ -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")