added CORS middleware

aritroCoder · Dec 11, 2023 · 440ba58 · 440ba58
1 parent 09aca35
commit 440ba58
Showing 1 changed file with 13 additions and 0 deletions.
diff --git a/forecaster/server.py b/forecaster/server.py
@@ -1,4 +1,5 @@
 from fastapi import FastAPI
+from fastapi.middleware.cors import CORSMiddleware
 from pydantic import BaseModel
 import pandas as pd
 import numpy as np
@@ -144,6 +145,18 @@ def validate_one_epoch(model, test_loader, loss_function):
 
 app = FastAPI()
 
+origins = [
+    "*",
+]
+
+app.add_middleware(
+    CORSMiddleware,
+    allow_origins=origins,
+    allow_credentials=True,
+    allow_methods=["*"],
+    allow_headers=["*"],
+)
+
 
 @app.get("/")
 async def root():