hysts HF staff commited on
Commit
811ffbe
1 Parent(s): edb2f4b

Add type annotation

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -63,8 +63,8 @@ default_style = styles[default_style_name]
63
  style_names = list(styles.keys())
64
 
65
 
66
- def apply_style(style, positive, negative=""):
67
- p, n = styles.get(style, default_style)
68
  return p.replace("{prompt}", positive), n + negative
69
 
70
 
 
63
  style_names = list(styles.keys())
64
 
65
 
66
+ def apply_style(style_name: str, positive: str, negative: str = "") -> tuple[str, str]:
67
+ p, n = styles.get(style_name, default_style)
68
  return p.replace("{prompt}", positive), n + negative
69
 
70