Tingman commited on
Commit
5febdd3
·
verified ·
1 Parent(s): 16ca2e3

Update gradio_app.py

Browse files
Files changed (1) hide show
  1. gradio_app.py +2 -2
gradio_app.py CHANGED
@@ -133,7 +133,8 @@ class MatchStereoDemo:
133
  }
134
 
135
  return size_presets.get(size_name, None)
136
-
 
137
  def process_images(self, left_image, right_image, mode, variant,
138
  low_res_init=False, inference_size_name="Original",
139
  precision="fp32", mat_impl="pytorch"):
@@ -253,7 +254,6 @@ examples = [
253
  ["examples/frame_0031_clean.png", "examples/frame_0032_clean.png", "flow", "base"],
254
  ]
255
 
256
- @spaces.GPU
257
  def process_inference(left_img, right_img, mode, variant,
258
  low_res_init, inference_size, precision, mat_impl):
259
  """Gradio function"""
 
133
  }
134
 
135
  return size_presets.get(size_name, None)
136
+
137
+ @spaces.GPU
138
  def process_images(self, left_image, right_image, mode, variant,
139
  low_res_init=False, inference_size_name="Original",
140
  precision="fp32", mat_impl="pytorch"):
 
254
  ["examples/frame_0031_clean.png", "examples/frame_0032_clean.png", "flow", "base"],
255
  ]
256
 
 
257
  def process_inference(left_img, right_img, mode, variant,
258
  low_res_init, inference_size, precision, mat_impl):
259
  """Gradio function"""