da

configs.py

@@ -1,3 +1,2 @@
-SECRET_KEY = 'secret_key_here'
 SQLALCHEMY_DATABASE_URI = 'sqlite:///users.db'
 SQLALCHEMY_TRACK_MODIFICATIONS = False

create_app.py

@@ -6,6 +6,7 @@ from transformers import LongformerTokenizer, AutoTokenizer, AutoModelForCausalL
 from models.model import *
 from utils.util_func import *
 from safetensors.torch import load_file
+import os 
 
 db = SQLAlchemy()
 login_manager = LoginManager()
@@ -37,12 +38,14 @@ def load_models():
 
 def create_app():
     set_seed(42)
-    load_models()
     app = Flask(__name__)
-    app.config.from_pyfile('configs.py')
+    app.secret_key = os.environ.get("SECRET_KEY", "this-should-be-overridden-in-prod")
+    app.config['SQLALCHEMY_DATABASE_URI'] = 'sqlite:///users.db'
+    app.config['SQLALCHEMY_TRACK_MODIFICATIONS'] = False
+    load_models()
     db.init_app(app)
-    login_manager.init_app(app)
     login_manager.login_view = 'auth.login'
+    login_manager.init_app(app)
     @login_manager.user_loader
     def load_user(user_id):
         return User.query.get(int(user_id))

inference/infer_single.py

@@ -25,7 +25,7 @@ def generate_and_score_essay(topic, essay):
     essay = replace_single_newlines(essay)
     paragraph_cnt = len(essay.replace('\\n\\n', '\\n').split('\\n'))
     gen_config = GenerationConfig(
-        max_new_tokens=512,     # cut way down from 1500
+        max_new_tokens=850,     # cut way down from 1500
         do_sample=True,
         top_k=50,
         top_p=0.9,

views/auth.py

@@ -34,11 +34,11 @@ def login():
     if request.method == 'POST':
         email = request.form['email']
         password = request.form['password']
-        print(email, password)
+        print(email, password, flush=True)
         user = User.query.filter_by(email=email).first()
-        print(user)
+        print(user, flush=True)
         if user and user.check_password(password):
-            print("LOGIN?")
+            print("LOGIN?", flush=True)
             login_user(user)
             
             return redirect(url_for('dashboard.dashboard'))