Spaces:

sayakpaul
/

fivek-retouching-maxim

Runtime error

sayakpaul HF staff commited on Oct 31, 2022

Commit

4dd0501

•

1 Parent(s): fe98fd9

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -12,8 +12,8 @@ from create_maxim_model import Model
 from maxim.configs import MAXIM_CONFIGS
 CKPT = "google/maxim-s2-enhancement-fivek"
-VARIANT = CKPT.split("/")[-1].split("_")[0]
-VARIANT = VARIANT[0].upper() + VARIANT[1]
 _MODEL = from_pretrained_keras(CKPT)

 from maxim.configs import MAXIM_CONFIGS
 CKPT = "google/maxim-s2-enhancement-fivek"
+VARIANT = CKPT.split("/")[-1].split("-")[1]
+VARIANT = VARIANT[0].upper() + "-" + VARIANT[1]
 _MODEL = from_pretrained_keras(CKPT)