Spaces:
Running
on
Zero
Running
on
Zero
Update main.py
Browse files
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 |
-
|
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",
|