diff --git a/requirements.txt b/requirements.txt index 820099c25da917f0c7ca2ec759effda9ae81a1a6..10a5b34a0239f84647ba66c386fd134a08d38847 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,4 +1,5 @@ Flask>=2.0.3 +flask-cors>=5.0.1 dotenv>=0.9.9 gradio_client>=1.7.0 pydantic>=1.8.2 diff --git a/src/main.py b/src/main.py index a63b8760bc60cf6b3db07399e8483efe0089faaf..9eae820e7403756360d4fe1f60b1f244fb307572 100644 --- a/src/main.py +++ b/src/main.py @@ -1,11 +1,16 @@ from dotenv import load_dotenv from flask import Flask +from flask_cors import CORS from routes.router import setup_routes load_dotenv() app = Flask(__name__) +CORS(app) + +app.config['CORS_HEADERS'] = 'Content-Type' + setup_routes(app) if __name__ == '__main__': diff --git a/src/routes/router.py b/src/routes/router.py index 128917a091b0f64b59166be4ea812d7088a21d5c..f5ece6bc883c1b9f3a35ed1e3bac16b692fcf077 100644 --- a/src/routes/router.py +++ b/src/routes/router.py @@ -11,7 +11,7 @@ glb_controller = GlbController() render_controller = RenderController() suggest_controller = SuggestController() -def setup_routes(app): +def setup_routes(app): @app.route('/glb', methods=['GET']) # Image to model def get_glb():