Update app.py

app.py

@@ -11,7 +11,7 @@ from PIL import Image
 from create_maxim_model import Model
 from maxim.configs import MAXIM_CONFIGS
 
-CKPT = "sayakpaul/S-2_deraining_raindrop"
+CKPT = "google/maxim-s2-deraining-raindrop"
 VARIANT = CKPT.split("/")[-1].split("_")[0]
 _MODEL = from_pretrained_keras(CKPT)
 
@@ -93,7 +93,7 @@ def infer(image):
 
 
 title = "Derain images containing rain drops or stripes."
-description = f"The underlying model is [this](https://huggingface.co/{CKPT}). You can use the model to derain images containing rain drops or stripes. To quickly try out the model, you can choose from the available sample images below, or you can submit your own image. Not that, internally, the model is re-initialized based on the spatial dimensions of the input image and this process is time-consuming."
+description = f"The underlying model is [this](https://huggingface.co/{CKPT}). You can use the model to derain images containing rain drops or stripes. This variant is particularly suited for draining image containing rain drops. If your images contain rain streaks, then use [this variant](https://huggingface.co/google/maxim-s2-deraining-rain13k). To quickly try out the model, you can choose from the available sample images below, or you can submit your own image. Not that, internally, the model is re-initialized based on the spatial dimensions of the input image and this process is time-consuming."
 
 iface = gr.Interface(
     infer,