Update app.py

app.py

@@ -7,19 +7,16 @@ from transformers import pipeline
 from transformers import AutoTokenizer, AutoModelForCausalLM
 
 # 下载模型
-base_dir = "/root/.cache/huggingface/hub"
-if not os.path.isdir(base_dir):
-    os.makedirs(base_dir)
-
-cmd_list = ["cd", "base_dir", "&&", "git clone", "https://gitee.com/lanzhiwang/gpt2.git", "models"]
-cmd_str = "".join(cmd_list)
-logging.info("cmd_str: %s", cmd_str)
-ret, out = subprocess.getstatusoutput(cmd_str)
-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")
+
+
+
+# Load model directly
+from transformers import AutoTokenizer, AutoModelForCausalLM
+
+tokenizer = AutoTokenizer.from_pretrained("gpt2")
+model = AutoModelForCausalLM.from_pretrained("gpt2")
+
+
 generator = pipeline('text-generation', model=model, tokenizer=tokenizer)
 
 # generator = pipeline('text-generation', model='gpt2')