-
Notifications
You must be signed in to change notification settings - Fork 1
/
Copy pathapp.py
35 lines (26 loc) · 889 Bytes
/
app.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
from flask import Flask
from flask_restful import Api
from flask_jwt import JWT
from flask_cors import CORS
from security import authenticate, identity
from resources.user import UserRegister
from resources.samplemodel import SampleModel
from resources.trainmodel import TrainModel
app = Flask(__name__)
CORS(app)
app.config['SQLALCHEMY_DATABASE_URI'] = 'sqlite:///data.db'
app.config['SQLALCHEMY_TRACK_MODIFICATIONS'] = False
app.config['PROPAGATE_EXCEPTIONS'] = True
app.secret_key = 'uvanshankar'
api = Api(app)
@app.before_first_request
def create_tables():
db.create_all()
jwt = JWT(app, authenticate, identity) # /auth
api.add_resource(UserRegister, '/register')
api.add_resource(SampleModel, '/model')
api.add_resource(TrainModel, '/trainmodel')
if __name__ == '__main__':
from db import db
db.init_app(app)
app.run(port=5000, threaded=False, debug=False)