Hugging Face's logo

Spaces:
hungdang1610
/
gopro-deblurring-maxim
Running

App Files Community
sayakpaul HF staff commited on
Commit
27081e5
•
1 Parent(s): 39cf748

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -12,8 +12,8 @@ from create_maxim_model import Model
12
  from maxim.configs import MAXIM_CONFIGS
13
 
14
  CKPT = "google/maxim-s3-deblurring-gopro"
15
- VARIANT = CKPT.split("/")[-1].split("_")[0]
16
- VARIANT = VARIANT[0].upper() + VARIANT[1]
17
  _MODEL = from_pretrained_keras(CKPT)
18
 
19
 
 
12
  from maxim.configs import MAXIM_CONFIGS
13
 
14
  CKPT = "google/maxim-s3-deblurring-gopro"
15
+ VARIANT = CKPT.split("/")[-1].split("-")[1]
16
+ VARIANT = VARIANT[0].upper() + "-" + VARIANT[1]
17
  _MODEL = from_pretrained_keras(CKPT)
18
 
19