prithivMLmods commited on
Commit
0f98551
·
verified ·
1 Parent(s): d193f0c

update app

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -580,7 +580,7 @@ def predictions_to_rrd(predictions, glbfile, target_dir, frame_filter="All", sho
580
  return rrd_path
581
 
582
 
583
- @spaces.GPU(duration=120)
584
  def run_model(target_dir, apply_mask=True, mask_edges=True, filter_black_bg=False, filter_white_bg=False):
585
  global model
586
  import torch
 
580
  return rrd_path
581
 
582
 
583
+ @spaces.GPU(duration=120, size="xlarge")
584
  def run_model(target_dir, apply_mask=True, mask_edges=True, filter_black_bg=False, filter_white_bg=False):
585
  global model
586
  import torch