cors updated

mlengineershub · May 20, 2024 · 986ab5e · 986ab5e
1 parent ac1eb29
commit 986ab5e
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 6 deletions.
diff --git a/endpoints/predict_cpu.py b/endpoints/predict_cpu.py
@@ -51,8 +51,8 @@
     CORSMiddleware,
     allow_origins=["*"],
     allow_credentials=True,
-    allow_methods=["GET", "POST"],
-    allow_headers=["*"]
+    allow_methods=["*"],
+    allow_headers=["*"],
 )
 
 model = AutoModelForImageClassification.from_pretrained(model_ckpt)

diff --git a/endpoints/predict_gpu.py b/endpoints/predict_gpu.py
@@ -51,8 +51,8 @@
     CORSMiddleware,
     allow_origins=["*"],
     allow_credentials=True,
-    allow_methods=["GET", "POST"],
-    allow_headers=["*"]
+    allow_methods=["*"],
+    allow_headers=["*"],
 )
 
 model = AutoModelForImageClassification.from_pretrained(model_ckpt)

diff --git a/endpoints/predict_mps.py b/endpoints/predict_mps.py
@@ -52,8 +52,8 @@
     CORSMiddleware,
     allow_origins=["*"],
     allow_credentials=True,
-    allow_methods=["GET", "POST"],
-    allow_headers=["*"]
+    allow_methods=["*"],
+    allow_headers=["*"],
 )
 
 model = AutoModelForImageClassification.from_pretrained(model_ckpt)