Spaces:
Paused
Paused
temp
Browse files
app.py
CHANGED
@@ -131,6 +131,9 @@ def demo_predict(input_blk, text, num_samples, steps, scale, seed, show_detail):
|
|
131 |
|
132 |
if __name__ == "__main__":
|
133 |
|
|
|
|
|
|
|
134 |
cfgs = OmegaConf.load("./configs/demo.yaml")
|
135 |
|
136 |
model = init_model(cfgs)
|
|
|
131 |
|
132 |
if __name__ == "__main__":
|
133 |
|
134 |
+
os.makedirs("./temp/attn_map", exist_ok=True)
|
135 |
+
os.makedirs("./temp/seg_map", exist_ok=True)
|
136 |
+
|
137 |
cfgs = OmegaConf.load("./configs/demo.yaml")
|
138 |
|
139 |
model = init_model(cfgs)
|