basic_auth

app.py

@@ -1,5 +1,7 @@
 from transformers import AutoModelForCausalLM, AutoTokenizer
 
+from flask_basicauth import BasicAuth
+
 model_name = "inclusionAI/Ring-mini-2.0"
 print(f"load model {model_name}")
 model = AutoModelForCausalLM.from_pretrained(
@@ -12,12 +14,23 @@ model = AutoModelForCausalLM.from_pretrained(
 print(f"load tokenizer {model_name}")
 tokenizer = AutoTokenizer.from_pretrained(model_name)
 
+
+
+# Чтобы защитить паролем конечную точку задания:
+# @basic_auth.required
 from flask import Flask  
 
 app = Flask(__name__)  
+
+basic_auth = BasicAuth(app)
+app.config['BASIC_AUTH_REALM'] = 'realm'
+app.config['BASIC_AUTH_USERNAME'] = 'vova'
+app.config['BASIC_AUTH_PASSWORD'] = 'vova'
+
 print(f"Flask app")
 # Главная страница
 @app.route("/", methods=['POST', 'GET']) #
+@basic_auth.required
 def root():
     if request.args.get('logs'): #logs=container  ?logs=container&__theme=system
         return f"<h1>Главная logs</h1>"