Spaces:
Running
on
Zero
Running
on
Zero
chong.zhang
commited on
Commit
·
2237355
1
Parent(s):
bac66d6
update
Browse files
app.py
CHANGED
@@ -191,13 +191,13 @@ if __name__ == '__main__':
|
|
191 |
lines = file.readlines()
|
192 |
updated_lines = []
|
193 |
for line in lines:
|
194 |
-
|
195 |
-
|
196 |
-
|
197 |
-
|
198 |
-
|
199 |
|
200 |
with open(yaml_file_path, 'w') as file:
|
201 |
-
|
202 |
|
203 |
main()
|
|
|
191 |
lines = file.readlines()
|
192 |
updated_lines = []
|
193 |
for line in lines:
|
194 |
+
if "basemodel_path: '../../pretrained_models/" in line:
|
195 |
+
line = line.replace('../../pretrained_models', 'pretrained_models')
|
196 |
+
elif "generator_path: '../../pretrained_models" in line:
|
197 |
+
line = line.replace('../../pretrained_models', 'pretrained_models')
|
198 |
+
updated_lines.append(line)
|
199 |
|
200 |
with open(yaml_file_path, 'w') as file:
|
201 |
+
file.writelines(updated_lines)
|
202 |
|
203 |
main()
|