Update main.py

main.py

@@ -9,27 +9,33 @@ import io
 from PIL import Image
 
 api = FastAPI()
-session = new_session()
+# Using birefnet-general for high quality background removal
+session = new_session("birefnet-general")
 
 @api.post("/api/remove-bg")
 async def remove_bg(file: UploadFile = File(...)):
         contents = await file.read()
+        # Process with BiRefNet
         output = remove(contents, session=session)
         return Response(content=output, media_type="image/png")
 
     @api.post("/api/enhance-photo")
 async def enhance_photo(file: UploadFile = File(...)):
         contents = await file.read()
+        # Remini style enhancement using Hugging Face Inference API (GFPGAN)
         API_URL = "https://api-inference.huggingface.co/models/TencentARC/GFPGAN"
-        response = requests.post(API_URL, data=contents)
-        if response.status_code != 200:
-                    return Response(content=contents, media_type="image/png")
-                return Response(content=response.content, media_type="image/png")
+
+        try:
+                    response = requests.post(API_URL, data=contents, timeout=30)
+                    if response.status_code == 200:
+                                    return Response(content=response.content, media_type="image/png")
+        else:
+                        # If API fails or is busy, return original image as fallback
+                        return Response(content=contents, media_type="image/png")
+except Exception as e:
+        # Fallback to original image on error
+        return Response(content=contents, media_type="image/png")
 
 @api.get("/")
 async def root():
-        return {"engine": "ALLBgremove Ultra Lite ACTIVE - Inference API Mode"}
-
-if __name__ == "__main__":
-        import uvicorn
-    uvicorn.run(api, host="0.0.0.0", port=7860)
+        return {"message": "All Background Remove API is running with BiRefNet and GFPGAN Enhancement"}