diff --git a/bird_cnn/server.py b/bird_cnn/server.py index 76f787d..863b181 100644 --- a/bird_cnn/server.py +++ b/bird_cnn/server.py @@ -41,10 +41,19 @@ model.load_state_dict(torch.load(full_path, map_location=torch.device(device))) model.to(device) model.eval() -app = FastAPI( - root_path="/cnn-api", - openapi_url="/cnn-api/openapi.json", - docs_url="/cnn-api/docs" +app = FastAPI() + +origins = [ + "https://marvinkrausser.com", + "https://api.marvinkrausser.com", +] + +app.add_middleware( + CORSMiddleware, + allow_origins=origins, + allow_credentials=True, + allow_methods=["*"], + allow_headers=["*"], ) @app.post("/predict") diff --git a/cnn_website/src/Bird_CNN.jsx b/cnn_website/src/Bird_CNN.jsx index 9241c35..ce7ac00 100644 --- a/cnn_website/src/Bird_CNN.jsx +++ b/cnn_website/src/Bird_CNN.jsx @@ -4,8 +4,8 @@ import './Bird_CNN.css'; function Bird_CNN() { const apiUrl = process.env.NODE_ENV === "development" - ? "https://marvinkrausser.com/cnn-api" - : "https://marvinkrausser.com/cnn-api"; + ? "https://api.marvinkrausser.com" + : "https://api.marvinkrausser.com"; const [file, setFile] = useState(null); const [birdClass, setBirdClass] = useState(null);