Update main.py

main.py

@@ -11,6 +11,7 @@ from src.routes.lora_training import lora_bp
 app = Flask(__name__, static_folder=os.path.join(os.path.dirname(__file__), 'static'))
 app.config['SECRET_KEY'] = 'asdf#FGSgvasgf$5$WGT'
 app.config['MAX_CONTENT_LENGTH'] = 500 * 1024 * 1024  # 500MB max file size
+flask==2.3.3
 
 app.register_blueprint(user_bp, url_prefix='/api')
 app.register_blueprint(lora_bp, url_prefix='/api')