增加报警机制

flask_predict_no_batch_t5.py

@@ -51,6 +51,33 @@ encoder, decoder, model, tokenizer = generatemodel.device_setup()
 autotitle = AutoTitle(encoder, decoder, model, tokenizer, start_id=0, end_id=tokenizer._token_end_id, maxlen=120)
 
 
+def smtp_f(name):
+    # 在下面的代码行中使用断点来调试脚本。
+    import smtplib
+    from email.mime.text import MIMEText
+    from email.header import Header
+
+    sender = '838878981@qq.com'  # 发送邮箱
+    receivers = ['838878981@qq.com']  # 接收邮箱
+    auth_code = "jfqtutaiwrtdbcge"  # 授权码
+
+    message = MIMEText('普通版降重项目出错，紧急', 'plain', 'utf-8')
+    message['From'] = Header("Sender<%s>" % sender)  # 发送者
+    message['To'] = Header("Receiver<%s>" % receivers[0])  # 接收者
+
+    subject = name
+    message['Subject'] = Header(subject, 'utf-8')
+
+    try:
+        server = smtplib.SMTP_SSL('smtp.qq.com', 465)
+        server.login(sender, auth_code)
+        server.sendmail(sender, receivers, message.as_string())
+        print("邮件发送成功")
+        server.close()
+    except smtplib.SMTPException:
+        print("Error: 无法发送邮件")
+
+
 class log:
     def __init__(self):
         pass
@@ -340,6 +367,7 @@ def classify():  # 调用模型，设置最大batch_size
 
 @app.route("/predict", methods=["POST"])
 def handle_query():
+    try:
         print(request.remote_addr)
         texts = request.json["texts"]
         text_type = request.json["text_type"]
@@ -362,6 +390,9 @@ def handle_query():
             print("ok")
         else:
             return_text = {"texts": "输入格式应该为字典", "probabilities": None, "status_code": 401}
+    except:
+        return_text = {"texts": "项目出错", "probabilities": None, "status_code": 402}
+        smtp_f("drop_weight_rewrite")
     return jsonify(return_text)  # 返回结果