Update main.py

This commit is contained in:
天涯古巷
2024-10-23 16:41:01 +08:00
committed by GitHub
parent b0ca454473
commit 3dc8f3bfe0

View File

@@ -1,3 +1,4 @@
import torch
import shutil import shutil
import subprocess import subprocess
import os import os
@@ -19,7 +20,6 @@ from backend.tools.inpaint_tools import create_mask, batch_generator
import importlib import importlib
import platform import platform
import tempfile import tempfile
import torch
import multiprocessing import multiprocessing
from shapely.geometry import Polygon from shapely.geometry import Polygon
import time import time