chong.zhang commited on
Commit
2237355
·
1 Parent(s): bac66d6
Files changed (1) hide show
  1. app.py +6 -6
app.py CHANGED
@@ -191,13 +191,13 @@ if __name__ == '__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()
 
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()