pookiefoof commited on
Commit
aa2a2cf
1 Parent(s): 8a9b9c8
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -77,6 +77,7 @@ class rm_bg_api:
77
  session_infer_path, providers=providers,
78
  )
79
 
 
80
  def remove_background(
81
  self,
82
  imgs: list[np.ndarray],
@@ -387,6 +388,7 @@ def main(
387
  infer3dapi = Inference3D_API()
388
  remove_api = rm_bg_api()
389
 
 
390
  def gen4views(image, width, height, seed, timestep, remove_bg):
391
  if remove_bg:
392
  image = remove_api.remove_background(
 
77
  session_infer_path, providers=providers,
78
  )
79
 
80
+ @spaces.GPU
81
  def remove_background(
82
  self,
83
  imgs: list[np.ndarray],
 
388
  infer3dapi = Inference3D_API()
389
  remove_api = rm_bg_api()
390
 
391
+ @spaces.GPU
392
  def gen4views(image, width, height, seed, timestep, remove_bg):
393
  if remove_bg:
394
  image = remove_api.remove_background(