niulx commited on
Commit
39680c0
1 Parent(s): e3ccd34

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +2 -2
main.py CHANGED
@@ -1,5 +1,5 @@
1
  import os
2
- #import spaces
3
  import torch
4
  import numpy as np
5
  import argparse
@@ -10,7 +10,7 @@ from utils import load_image, load_mask, load_mask_edit
10
  from utils_mask import process_mask_move_torch, process_mask_remove_torch, mask_union_torch, mask_substract_torch, create_outer_edge_mask_torch
11
  from utils_mask import check_mask_overlap_torch, check_cover_all_torch, visualize_mask_list, get_mask_difference_torch, save_mask_list_to_npys
12
 
13
- #@spaces.GPU(duration=45)
14
 
15
  def run_main(
16
  name="example_tmp",
 
1
  import os
2
+ import spaces
3
  import torch
4
  import numpy as np
5
  import argparse
 
10
  from utils_mask import process_mask_move_torch, process_mask_remove_torch, mask_union_torch, mask_substract_torch, create_outer_edge_mask_torch
11
  from utils_mask import check_mask_overlap_torch, check_cover_all_torch, visualize_mask_list, get_mask_difference_torch, save_mask_list_to_npys
12
 
13
+ @spaces.GPU(duration=45)
14
 
15
  def run_main(
16
  name="example_tmp",