ZenosArrows commited on
Commit
d94ba65
·
verified ·
1 Parent(s): f9924d6

Fix ZeroGPU

Browse files
Files changed (1) hide show
  1. app.py +1 -0
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])