Spaces:
Sleeping
Sleeping
push that shiii
Browse files
app.py
CHANGED
|
@@ -3,6 +3,7 @@ from fastapi.responses import Response, JSONResponse
|
|
| 3 |
from PIL import Image
|
| 4 |
import io
|
| 5 |
import logging
|
|
|
|
| 6 |
from bg_remover import BackgroundRemover
|
| 7 |
|
| 8 |
logging.basicConfig(level=logging.INFO)
|
|
@@ -14,9 +15,9 @@ app = FastAPI(
|
|
| 14 |
version="1.0.0"
|
| 15 |
)
|
| 16 |
|
| 17 |
-
background_remover: BackgroundRemover = None
|
| 18 |
|
| 19 |
-
def get_background_remover():
|
| 20 |
global background_remover
|
| 21 |
if background_remover is None:
|
| 22 |
logger.info("Initializing background remover...")
|
|
@@ -152,3 +153,6 @@ async def remove_background_json(
|
|
| 152 |
}
|
| 153 |
)
|
| 154 |
|
|
|
|
|
|
|
|
|
|
|
|
| 3 |
from PIL import Image
|
| 4 |
import io
|
| 5 |
import logging
|
| 6 |
+
from typing import Optional
|
| 7 |
from bg_remover import BackgroundRemover
|
| 8 |
|
| 9 |
logging.basicConfig(level=logging.INFO)
|
|
|
|
| 15 |
version="1.0.0"
|
| 16 |
)
|
| 17 |
|
| 18 |
+
background_remover: Optional[BackgroundRemover] = None
|
| 19 |
|
| 20 |
+
def get_background_remover() -> BackgroundRemover:
|
| 21 |
global background_remover
|
| 22 |
if background_remover is None:
|
| 23 |
logger.info("Initializing background remover...")
|
|
|
|
| 153 |
}
|
| 154 |
)
|
| 155 |
|
| 156 |
+
if __name__ == "__main__":
|
| 157 |
+
import uvicorn
|
| 158 |
+
uvicorn.run(app, host="0.0.0.0", port=7860)
|