Spaces:
Running
on
Zero
Running
on
Zero
progress bar fixed
Browse files
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 |
|