Spaces:
Build error
Build error
Update app.py
Browse files
app.py
CHANGED
|
@@ -11,7 +11,7 @@ from PIL import Image
|
|
| 11 |
from create_maxim_model import Model
|
| 12 |
from maxim.configs import MAXIM_CONFIGS
|
| 13 |
|
| 14 |
-
CKPT = "sayakpaul/S-
|
| 15 |
VARIANT = CKPT.split("/")[-1].split("_")[0]
|
| 16 |
_MODEL = from_pretrained_keras(CKPT)
|
| 17 |
|
|
|
|
| 11 |
from create_maxim_model import Model
|
| 12 |
from maxim.configs import MAXIM_CONFIGS
|
| 13 |
|
| 14 |
+
CKPT = "sayakpaul/S-2_deraining_raindrop"
|
| 15 |
VARIANT = CKPT.split("/")[-1].split("_")[0]
|
| 16 |
_MODEL = from_pretrained_keras(CKPT)
|
| 17 |
|