tosei0000 commited on
Commit
29840c0
·
verified ·
1 Parent(s): b68b228

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -2,7 +2,7 @@ from transformers import AutoTokenizer, AutoModelForCausalLM
2
  import torch
3
 
4
  # 设置模型路径(替换为你上传后显示的路径名)
5
- model_path = "tosei0000/code-AI" # 修改为你上传的文件夹名
6
 
7
  # 加载模型和分词器
8
  tokenizer = AutoTokenizer.from_pretrained(model_path, trust_remote_code=True)
 
2
  import torch
3
 
4
  # 设置模型路径(替换为你上传后显示的路径名)
5
+ model_path = "tosei0000/code" # 修改为你上传的文件夹名
6
 
7
  # 加载模型和分词器
8
  tokenizer = AutoTokenizer.from_pretrained(model_path, trust_remote_code=True)