Update app.py

app.py

@@ -7,7 +7,7 @@ from transformers import pipeline
 from transformers import AutoTokenizer, AutoModelForCausalLM
 
 # 下载模型
-base_dir = "/root/.cache/huggingface/hub"
+base_dir = ".cache/huggingface/hub"
 if not os.path.isdir(base_dir):
     os.makedirs(base_dir)
 
@@ -18,8 +18,8 @@ ret, out = subprocess.getstatusoutput(cmd_str)
 print("ret:", ret)
 print("out:", 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")
+tokenizer = AutoTokenizer.from_pretrained(pretrained_model_name_or_path=".cache/huggingface/hub/models")
+model = AutoModelForCausalLM.from_pretrained(pretrained_model_name_or_path=".cache/huggingface/hub/models")
 generator = pipeline('text-generation', model=model, tokenizer=tokenizer)
 
 # generator = pipeline('text-generation', model='gpt2')