Update app.py

app.py

@@ -1,11 +1,9 @@
 import io
 import os
 import threading
-import queue
-import psutil
-import asyncio
 from fastapi import FastAPI, File, UploadFile, Request
 from fastapi.responses import StreamingResponse, HTMLResponse
+from fastapi.staticfiles import StaticFiles
 from PIL import Image
 import torch
 import torchvision.transforms as transforms
@@ -33,19 +31,7 @@ print(f"ONNX model loaded with providers: {providers}")
 # Lock for thread-safe ONNX inference
 onnx_lock = threading.Lock()
 
-# Request queue for load management
-request_queue = queue.Queue()
-
-# System thresholds
-MAX_CPU_USAGE = 85.0  # percent
-MAX_RAM_USAGE = 85.0  # percent
-
-def system_under_load() -> bool:
-    """Check if system is under heavy load."""
-    cpu = psutil.cpu_percent(interval=0.1)
-    mem = psutil.virtual_memory().percent
-    return cpu > MAX_CPU_USAGE or mem > MAX_RAM_USAGE
-
+# Helper functions
 def run_model_onnx(input_tensor: torch.Tensor) -> torch.Tensor:
     with onnx_lock:  # ensure thread safety
         ort_inputs = {onnx_session.get_inputs()[0].name: input_tensor.cpu().numpy()}
@@ -67,14 +53,10 @@ def process_image(image: Image.Image) -> Image.Image:
     return image_rgba
 
 # FastAPI app
-app = FastAPI(title="Background Removal API with Auto Resource Management")
+app = FastAPI(title="Background Removal API")
 
 @app.post("/remove-background")
 async def remove_background(file: UploadFile = File(...)):
-    # Queue request if system is under load
-    while system_under_load():
-        await asyncio.sleep(0.2)
-
     image = Image.open(file.file).convert("RGB")
     result_image = process_image(image)
 
@@ -177,4 +159,4 @@ async def home(request: Request):
         </script>
     </body>
     </html>
-    """
+    """