sayakpaul HF staff commited on
Commit
fe98fd9
1 Parent(s): 8750d72

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -13,7 +13,7 @@ from maxim.configs import MAXIM_CONFIGS
13
 
14
  CKPT = "google/maxim-s2-enhancement-fivek"
15
  VARIANT = CKPT.split("/")[-1].split("_")[0]
16
- VARIANT = VARIANT.upper()
17
  _MODEL = from_pretrained_keras(CKPT)
18
 
19
 
 
13
 
14
  CKPT = "google/maxim-s2-enhancement-fivek"
15
  VARIANT = CKPT.split("/")[-1].split("_")[0]
16
+ VARIANT = VARIANT[0].upper() + VARIANT[1]
17
  _MODEL = from_pretrained_keras(CKPT)
18
 
19