Spaces:
Runtime error
Runtime error
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 |
|