Spaces:
Build error
Build error
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)
|