Update app.py

app.py

@@ -9,7 +9,6 @@ from io import BytesIO
 from PIL import Image, ImageFilter, ImageEnhance
 import rembg
 import onnxruntime as ort
-from functools import lru_cache
 
 app = FastAPI()
 
@@ -18,11 +17,6 @@ options = ort.SessionOptions()
 options.intra_op_num_threads = 2  # Limita o número de threads para evitar sobrecarga
 options.execution_mode = ort.ExecutionMode.ORT_SEQUENTIAL  # Execução sequencial para melhor desempenho em CPU
 
-# Cache do modelo u2net (carrega apenas uma vez)
-@lru_cache(maxsize=1)
-def load_u2net_model():
-    return rembg.get_model("u2net")
-
 def resize_image(image, max_size=512):
     """Redimensiona a imagem para uma largura máxima de 512px, mantendo a proporção."""
     width, height = image.size
@@ -55,8 +49,7 @@ async def remove_background(image_url: str):
         image = adjust_brightness_contrast(image)
         
         # Remove o fundo da imagem usando rembg
-        model = load_u2net_model()  # Carrega o modelo (usando cache)
-        output = rembg.remove(image, model=model, session_options=options)
+        output = rembg.remove(image, session_options=options)
         
         # Pós-processamento: suaviza as bordas
         output = output.filter(ImageFilter.SMOOTH_MORE)