sayakpaul HF staff commited on
Commit
2ed2741
·
1 Parent(s): 2ecc820

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -11,8 +11,8 @@ from PIL import Image
11
  from create_maxim_model import Model
12
  from maxim.configs import MAXIM_CONFIGS
13
 
14
- CKPT = "sayakpaul/S-2_dehazing_sots-outdoor"
15
- VARIANT = CKPT.split("/")[-1].split("_")[0]
16
  _MODEL = from_pretrained_keras(CKPT)
17
 
18
 
@@ -93,7 +93,7 @@ def infer(image):
93
 
94
 
95
  title = "Dehaze hazy images."
96
- description = f"The underlying model is [this](https://huggingface.co/{CKPT}). You can use the model to dehaze hazy images. There is [another version of the model]('https://hf.co/sayakpaul/S-2_dehazing_sots-indoor') you can try out. 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."
97
 
98
  iface = gr.Interface(
99
  infer,
 
11
  from create_maxim_model import Model
12
  from maxim.configs import MAXIM_CONFIGS
13
 
14
+ CKPT = "google/maxim-s2-dehazing-sots-outdoor"
15
+ VARIANT = CKPT.split("/")[-1].split("-")[1]
16
  _MODEL = from_pretrained_keras(CKPT)
17
 
18
 
 
93
 
94
 
95
  title = "Dehaze hazy images."
96
+ description = f"The underlying model is [this](https://huggingface.co/{CKPT}). You can use the model to dehaze hazy images. There is [another version of the model]('https://hf.co/google/maxim-s2-dehazing-sots-indoor') (better suited for indoor images) you can try out. 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."
97
 
98
  iface = gr.Interface(
99
  infer,