tori29umai commited on
Commit
48b19ab
1 Parent(s): f83e5e8
Files changed (2) hide show
  1. app.py +2 -0
  2. requirements.txt +2 -1
app.py CHANGED
@@ -21,7 +21,9 @@ from utils.prompt_analysis import PromptAnalysis
21
 
22
  path = os.getcwd()
23
  cn_dir = f"{path}/controlnet"
 
24
  tagger_dir = f"{path}/tagger"
 
25
 
26
  load_cn_model(cn_dir)
27
  load_cn_config(cn_dir)
 
21
 
22
  path = os.getcwd()
23
  cn_dir = f"{path}/controlnet"
24
+ os.makedirs(cn_dir)
25
  tagger_dir = f"{path}/tagger"
26
+ os.mkdir(tagger_dir)
27
 
28
  load_cn_model(cn_dir)
29
  load_cn_config(cn_dir)
requirements.txt CHANGED
@@ -20,4 +20,5 @@ spaces
20
  torch==2.2
21
  controlnet-aux==0.0.9
22
  onnx
23
- onnxruntime
 
 
20
  torch==2.2
21
  controlnet-aux==0.0.9
22
  onnx
23
+ onnxruntime
24
+ mediapipe