优化

app.py

@@ -1,28 +1,29 @@
+import os, shutil
 from zipfile import ZipFile
 import numpy as np
-
-import os, shutil
-
 import gradio as gr
 from PIL import Image
-import sys, os
 from rembg import new_session, remove
-from utils import functions
+from logging import getLogger, StreamHandler, DEBUG
 from utils.functions import complete, clean, clean_by_name, get_random_name
 
-import os, sys
-import argparse
-
-from logging import getLogger, StreamHandler, DEBUG
 logger = getLogger(__name__)
-handler = StreamHandler(); handler.setLevel(DEBUG)
+handler = StreamHandler()
+handler.setLevel(DEBUG)
 logger.setLevel(DEBUG)
 logger.addHandler(handler)
 logger.propagate = False
-
-
         
 def run_rembg(img):
+    """
+    Remove background from an image using U2-Net algorithm.
+
+    Args:
+        img: numpy array, input image data.
+
+    Returns:
+        A cropped PIL image object.
+    """
     output = remove(img)
     output_pil = Image.fromarray(output)
 
@@ -45,6 +46,15 @@ def run_rembg_withmodel(img, model):
     return output_pil
     
 def from_zip(inputs):
+    """
+    Read images from a zip file and output a processed zip file.
+
+    Args:
+        inputs: list of file objects, input zip file.
+
+    Returns:
+        A tuple of output zip file name and a completion message.
+    """
     work_dir = get_random_name()
     os.makedirs(work_dir, exist_ok=True)