kalhdrawi commited on
Commit
558c02f
·
1 Parent(s): 3bf1031

إضافة ملفات جديدة

Browse files
Files changed (1) hide show
  1. server.py +9 -8
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)