Update app.py

app.py

@@ -7,6 +7,8 @@ from optimum.onnxruntime import ORTModelForCausalLM
 
 import itertools
 import regex as re
+import logging
+
 
 
 user_token = "<User>"
@@ -14,6 +16,15 @@ eos_token = "<EOS>"
 bos_token = "<BOS>"
 bot_token = "<Assistant>"
 
+
+logger = logging.getLogger()
+handler = logging.StreamHandler()
+formatter = logging.Formatter(
+        '%(asctime)s %(name)-12s %(levelname)-8s %(message)s')
+handler.setFormatter(formatter)
+logger.addHandler(handler)
+logger.setLevel(logging.INFO)
+
 max_context_length = 750
 
 def format(history):
@@ -25,7 +36,7 @@ def format(history):
         else:
             prompt += f"{bot_token}{txt}"
     prompt += bot_token
-    print(prompt)
+    logger.info(prompt)
     return prompt
 
 def remove_spaces_between_chinese(text):
@@ -133,9 +144,9 @@ def gradio(model, tokenizer):
             undo_btn = "↩️ Remove last turn",
             clear_btn = "➕ New conversation",
             examples=[
-                ["给我5个英文名。", 100, 0.9, 0.95, 1.2, 1],
-                ["给我讲一个笑话。", 50, 0.8, 0.9, 1.3, 6],
-                ["Describe spring in english.", 50, 0.9, 1.0, 1, 5]
+                ["写一篇介绍人工智能的文章。", 30, 0.9, 0.95, 1.2, 5],
+                ["给我讲一个笑话。", 50, 0.8, 0.9, 1.2, 6],
+                ["Can you describe spring in English?", 50, 0.9, 1.0, 1, 5]
             ]
         )