TruVlad commited on
Commit
3f252cd
·
verified ·
1 Parent(s): c9da9ce

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +11 -1
app.py CHANGED
@@ -1,6 +1,7 @@
1
  from transformers import AutoModelForCausalLM, AutoTokenizer
2
-
3
  from flask_basicauth import BasicAuth
 
4
 
5
  model_name = "inclusionAI/Ring-mini-2.0"
6
  print(f"load model {model_name}")
@@ -14,7 +15,16 @@ model = AutoModelForCausalLM.from_pretrained(
14
  print(f"load tokenizer {model_name}")
15
  tokenizer = AutoTokenizer.from_pretrained(model_name)
16
 
 
 
 
 
 
 
 
17
 
 
 
18
 
19
  # Чтобы защитить паролем конечную точку задания:
20
  # @basic_auth.required
 
1
  from transformers import AutoModelForCausalLM, AutoTokenizer
2
+ #Flask-APScheduler
3
  from flask_basicauth import BasicAuth
4
+ from datatime import datatime
5
 
6
  model_name = "inclusionAI/Ring-mini-2.0"
7
  print(f"load model {model_name}")
 
15
  print(f"load tokenizer {model_name}")
16
  tokenizer = AutoTokenizer.from_pretrained(model_name)
17
 
18
+ from flask_apscheduler import APScheduler
19
+
20
+ app = Flask(__name__)
21
+ scheduler = APScheduler()
22
+
23
+ def scheduleTask():
24
+ print(f"This test runs every 30 seconds {datatime.now()}")
25
 
26
+ scheduler.add_job(id = 'Scheduled Task', func=scheduleTask, trigger="interval", seconds=30)
27
+ scheduler.start()
28
 
29
  # Чтобы защитить паролем конечную точку задания:
30
  # @basic_auth.required