transformers

app.py

@@ -1,3 +1,15 @@
+from transformers import AutoModelForCausalLM, AutoTokenizer
+
+model_name = "inclusionAI/Ring-mini-2.0"
+
+model = AutoModelForCausalLM.from_pretrained(
+    model_name,
+    torch_dtype="auto",
+    device_map="auto",
+    trust_remote_code=True
+)
+tokenizer = AutoTokenizer.from_pretrained(model_name)
+
 from flask import Flask  
 
 app = Flask(__name__)  
@@ -5,7 +17,7 @@ app = Flask(__name__)
 # Главная страница
 @app.route("/")
 def home():
-    return "<h1>Главная страница</h1><p>Добро пожаловать!</p>"
+    return f"<h1>Главная страница</h1><p>Добро пожаловать!</p><p>current model {model_name}</p>"
 
 # Страница "О нас"
 @app.route("/about")