Ashish Ajin commited on
Commit
d3bfd18
·
1 Parent(s): cdf1f46

Fix optimized letter case

Browse files
Files changed (1) hide show
  1. wai_service.py +3 -1
wai_service.py CHANGED
@@ -93,7 +93,9 @@ def handler(payload: dict) -> str:
93
  # sensible defaults
94
  cfg.render_size = getattr(cfg, "render_size", 384) # <= set here
95
  cfg.word = cfg.word.upper()
96
- cfg.optimized_letter = getattr(cfg, "optimized_letter", cfg.word[-1])
 
 
97
  if getattr(cfg.diffusion, "model", ...) is Ellipsis:
98
  cfg.diffusion.model = "runwayml/stable-diffusion-v1-5"
99
 
 
93
  # sensible defaults
94
  cfg.render_size = getattr(cfg, "render_size", 384) # <= set here
95
  cfg.word = cfg.word.upper()
96
+ cfg.optimized_letter = getattr(cfg, "optimized_letter", cfg.word[-1]).upper()
97
+ cfg.letter = f"{cfg.font}_{cfg.optimized_letter}_scaled"
98
+ cfg.target = f"code/data/init/{cfg.letter}"
99
  if getattr(cfg.diffusion, "model", ...) is Ellipsis:
100
  cfg.diffusion.model = "runwayml/stable-diffusion-v1-5"
101