alexnasa commited on
Commit
e794d57
·
verified ·
1 Parent(s): 277e783

progress bar fixed

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -179,7 +179,7 @@ def preprocess_image(input_image: Image.Image) -> Image.Image:
179
  return img
180
 
181
  @spaces.GPU()
182
- def preprocess_n_magnify(input_image: Image.Image):
183
  """
184
  Preprocess the input image and perform a single-step 4× magnification using the SeeSR pipeline.
185
 
@@ -197,7 +197,7 @@ def preprocess_n_magnify(input_image: Image.Image):
197
 
198
  processed_img = preprocess_image(input_image)
199
 
200
- img, magnified_img = magnify(processed_img)
201
 
202
  return (img, magnified_img)
203
 
 
179
  return img
180
 
181
  @spaces.GPU()
182
+ def preprocess_n_magnify(input_image: Image.Image, progress=gr.Progress(track_tqdm=True),):
183
  """
184
  Preprocess the input image and perform a single-step 4× magnification using the SeeSR pipeline.
185
 
 
197
 
198
  processed_img = preprocess_image(input_image)
199
 
200
+ img, magnified_img = magnify(processed_img, progress=progress)
201
 
202
  return (img, magnified_img)
203