Spaces:

sayakpaul
/

fivek-retouching-maxim

Runtime error

sayakpaul HF Staff commited on Oct 31, 2022

Commit

8750d72

1 Parent(s): ef31f77

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -11,8 +11,9 @@ from PIL import Image
 from create_maxim_model import Model
 from maxim.configs import MAXIM_CONFIGS
-CKPT = "sayakpaul/S-2_enhancement_fivek"
 VARIANT = CKPT.split("/")[-1].split("_")[0]
 _MODEL = from_pretrained_keras(CKPT)

 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.upper()
 _MODEL = from_pretrained_keras(CKPT)