From 3dc8f3bfe0ca3f5222d32d8c7356d9e16c8233b2 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E5=A4=A9=E6=B6=AF=E5=8F=A4=E5=B7=B7?= Date: Wed, 23 Oct 2024 16:41:01 +0800 Subject: [PATCH] Update main.py --- backend/main.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/backend/main.py b/backend/main.py index b0de01d..3aa0a23 100644 --- a/backend/main.py +++ b/backend/main.py @@ -1,3 +1,4 @@ +import torch import shutil import subprocess import os @@ -19,7 +20,6 @@ from backend.tools.inpaint_tools import create_mask, batch_generator import importlib import platform import tempfile -import torch import multiprocessing from shapely.geometry import Polygon import time