cherubicxn commited on
Commit
1d759ae
·
1 Parent(s): e33fa08
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -32,7 +32,8 @@ def stop_run():
32
  gr.update(visible=False),
33
  )
34
 
35
- @spaces.GPU()
 
36
  def process_image(
37
  input_image,
38
  model_name='scalelsd-vitbase-v2-train-sa1b.pt',
 
32
  gr.update(visible=False),
33
  )
34
 
35
+ # @spaces.GPU()
36
+ @spaces.GPU
37
  def process_image(
38
  input_image,
39
  model_name='scalelsd-vitbase-v2-train-sa1b.pt',