Spaces:
Running
on
Zero
Running
on
Zero
tori29umai
commited on
Commit
•
48b19ab
1
Parent(s):
f83e5e8
app.py
Browse files- app.py +2 -0
- 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
|