Spaces:
Running
on
Zero
Running
on
Zero
Fix ZeroGPU
Browse files
app.py
CHANGED
|
@@ -121,6 +121,7 @@ def upscale_image(image, model, background, discard_alpha):
|
|
| 121 |
image = model.infer(image)
|
| 122 |
return image.convert("RGB") if discard_alpha else image
|
| 123 |
|
|
|
|
| 124 |
def on_submit(image, batch_images, book, encoder, upscale_model, upscale_method, denoise_level, discard_alpha, progress=gr.Progress()):
|
| 125 |
model_name = encoder2name[encoder]
|
| 126 |
model = DepthAnythingV2(**model_configs[encoder])
|
|
|
|
| 121 |
image = model.infer(image)
|
| 122 |
return image.convert("RGB") if discard_alpha else image
|
| 123 |
|
| 124 |
+
@spaces.GPU
|
| 125 |
def on_submit(image, batch_images, book, encoder, upscale_model, upscale_method, denoise_level, discard_alpha, progress=gr.Progress()):
|
| 126 |
model_name = encoder2name[encoder]
|
| 127 |
model = DepthAnythingV2(**model_configs[encoder])
|