Spaces:

sayakpaul
/

raindrop-deraining-maxim

Build error

sayakpaul HF Staff commited on Oct 31, 2022

Commit

aff0e60

1 Parent(s): e1c5111

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -13,7 +13,7 @@ from maxim.configs import MAXIM_CONFIGS
 CKPT = "google/maxim-s2-deraining-raindrop"
 VARIANT = CKPT.split("/")[-1].split("_")[0]
-VARIANT = VARIANT.upper()
 _MODEL = from_pretrained_keras(CKPT)

 CKPT = "google/maxim-s2-deraining-raindrop"
 VARIANT = CKPT.split("/")[-1].split("_")[0]
+VARIANT = VARIANT[0].upper() + VARIANT[1]
 _MODEL = from_pretrained_keras(CKPT)