TruVlad commited on
Commit
fd5930b
·
verified ·
1 Parent(s): 6011586
Files changed (1) hide show
  1. app.py +35 -0
app.py ADDED
@@ -0,0 +1,35 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ from transformers import AutoModelForCausalLM, AutoTokenizer
2
+
3
+ model_name = "inclusionAI/Ring-mini-2.0"
4
+ print(f"load model {model_name}")
5
+ model = AutoModelForCausalLM.from_pretrained(
6
+ model_name,
7
+ torch_dtype="auto",
8
+ device_map="auto",
9
+ trust_remote_code=True
10
+ )
11
+
12
+ print(f"load tokenizer {model_name}")
13
+ tokenizer = AutoTokenizer.from_pretrained(model_name)
14
+
15
+ from flask import Flask
16
+
17
+ app = Flask(__name__)
18
+ print(f"Flask app")
19
+ # Главная страница
20
+ @app.route("/")
21
+ def home():
22
+ return f"<h1>Главная страница</h1><p>Добро пожаловать!</p><p>current model {model_name}</p>"
23
+
24
+ # Страница "О нас"
25
+ @app.route("/about")
26
+ def about():
27
+ return "<h1>О нас</h1><p>Мы изучаем Flask!</p>"
28
+
29
+ # Страница "Контакты"
30
+ @app.route("/contact")
31
+ def contact():
32
+ return "<h1>Контакты</h1><p>Свяжитесь с нами: email@example.com</p>"
33
+
34
+ if __name__ == "__main__":
35
+ app.run(debug=False, host='0.0.0.0', port=7860)