Spaces:
Runtime error
Runtime error
إضافة ملفات جديدة
Browse files
server.py
CHANGED
|
@@ -13,14 +13,6 @@ from fastapi.middleware.cors import CORSMiddleware
|
|
| 13 |
from PIL import Image
|
| 14 |
|
| 15 |
from briarmbg import BriaRMBG
|
| 16 |
-
# Enable CORS for external API usage (adjust origins as needed)
|
| 17 |
-
app.add_middleware(
|
| 18 |
-
CORSMiddleware,
|
| 19 |
-
allow_origins=["*"],
|
| 20 |
-
allow_credentials=True,
|
| 21 |
-
allow_methods=["*"],
|
| 22 |
-
allow_headers=["*"],
|
| 23 |
-
)
|
| 24 |
|
| 25 |
|
| 26 |
app = FastAPI(title="BRIA RMBG Service", version="0.1.0")
|
|
@@ -32,6 +24,15 @@ net: Optional[BriaRMBG] = None
|
|
| 32 |
_device: Optional[torch.device] = None
|
| 33 |
|
| 34 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 35 |
def _resize_image(image: Image.Image) -> Image.Image:
|
| 36 |
image = image.convert("RGB")
|
| 37 |
model_input_size = (1024, 1024)
|
|
|
|
| 13 |
from PIL import Image
|
| 14 |
|
| 15 |
from briarmbg import BriaRMBG
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 16 |
|
| 17 |
|
| 18 |
app = FastAPI(title="BRIA RMBG Service", version="0.1.0")
|
|
|
|
| 24 |
_device: Optional[torch.device] = None
|
| 25 |
|
| 26 |
|
| 27 |
+
# Enable CORS for external API usage (adjust origins as needed)
|
| 28 |
+
app.add_middleware(
|
| 29 |
+
CORSMiddleware,
|
| 30 |
+
allow_origins=["*"],
|
| 31 |
+
allow_credentials=True,
|
| 32 |
+
allow_methods=["*"],
|
| 33 |
+
allow_headers=["*"],
|
| 34 |
+
)
|
| 35 |
+
|
| 36 |
def _resize_image(image: Image.Image) -> Image.Image:
|
| 37 |
image = image.convert("RGB")
|
| 38 |
model_input_size = (1024, 1024)
|