shotx-deraining-maxim

Running

sayakpaul HF staff commited on Oct 31, 2022

Commit

093a9b4

•

1 Parent(s): aff0e60

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-deraining-raindrop"
-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-deraining-raindrop"
+VARIANT = CKPT.split("/")[-1].split("-")[1]
+VARIANT = VARIANT[0].upper() + "-" + VARIANT[1]
 _MODEL = from_pretrained_keras(CKPT)