logging

app.py

@@ -1,5 +1,7 @@
 import os
 import subprocess
+import logging
+
 import gradio as gr
 from transformers import pipeline
 from transformers import AutoTokenizer, AutoModelForCausalLM
@@ -11,10 +13,10 @@ if not os.path.isdir(base_dir):
 
 cmd_list = ["cd", "base_dir", "&&", "git clone", "https://gitee.com/lanzhiwang/gpt2.git", "models"]
 cmd_str = "".join(cmd_list)
-print("cmd_str:", cmd_str)
+logging.info("cmd_str: %s", cmd_str)
 ret, out = subprocess.getstatusoutput(cmd_str)
-print("ret:", ret)
-print("out:", out)
+logging.info("ret: %s", ret)
+logging.info("out: %s", out)
 
 tokenizer = AutoTokenizer.from_pretrained(pretrained_model_name_or_path="/root/.cache/huggingface/hub/models")
 model = AutoModelForCausalLM.from_pretrained(pretrained_model_name_or_path="/root/.cache/huggingface/hub/models")