Spaces:
Running
Running
Update lama_inpaint.py
Browse files- lama_inpaint.py +3 -3
lama_inpaint.py
CHANGED
@@ -17,9 +17,9 @@ os.environ['NUMEXPR_NUM_THREADS'] = '1'
|
|
17 |
|
18 |
sys.path.insert(0, str(Path(__file__).resolve().parent / "third_party" / "lama"))
|
19 |
|
20 |
-
from saicinpainting.evaluation.utils import move_to_device
|
21 |
-
from saicinpainting.training.trainers import load_checkpoint
|
22 |
-
from saicinpainting.evaluation.data import pad_tensor_to_modulo
|
23 |
|
24 |
from utils import load_img_to_array, save_array_to_img
|
25 |
|
|
|
17 |
|
18 |
sys.path.insert(0, str(Path(__file__).resolve().parent / "third_party" / "lama"))
|
19 |
|
20 |
+
from lama.saicinpainting.evaluation.utils import move_to_device
|
21 |
+
from lama.saicinpainting.training.trainers import load_checkpoint
|
22 |
+
from lama.saicinpainting.evaluation.data import pad_tensor_to_modulo
|
23 |
|
24 |
from utils import load_img_to_array, save_array_to_img
|
25 |
|