jiaweir commited on
Commit
20167fb
β€’
1 Parent(s): c54a4cd
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -60,7 +60,7 @@ def optimize_stage_1(image_block: Image.Image, preprocess_chk: bool, seed_slider
60
  # return [os.path.join('logs', 'tmp_rgba_model.ply')]
61
  return os.path.join('tmp_data', f'{img_hash}_rgba_generated.mp4')
62
 
63
- @spaces.GPU(duration=180)
64
  def optimize_stage_2(image_block: Image.Image, seed_slider: int):
65
  img_hash = hashlib.sha256(image_block.tobytes()).hexdigest()
66
 
 
60
  # return [os.path.join('logs', 'tmp_rgba_model.ply')]
61
  return os.path.join('tmp_data', f'{img_hash}_rgba_generated.mp4')
62
 
63
+ @spaces.GPU(duration=120)
64
  def optimize_stage_2(image_block: Image.Image, seed_slider: int):
65
  img_hash = hashlib.sha256(image_block.tobytes()).hexdigest()
66