Register gemma blueprints

1 files changed, 5 insertions, 10 deletions

diff --git a/main.py b/main.py
index 33d3882..8b2fa73 100644
--- a/main.py
+++ b/main.py
@@ -1,14 +1,9 @@
 """Main entry point for project"""
 
-from flask import Flask
-app = Flask(__name__)
+from gn3.app import create_app
+from gn3.api.gemma import gemma
 
 
-@app.route("/")
-def hello():
-    """Test hello world"""
-    return "Hello World!"
-
-
-if __name__ == '__main__':
-    app.run()
+app = create_app()
+app.register_blueprint(gemma, url_prefix="/gemma")
+app.run(host="0.0.0.0")