Update README.md
Browse files
README.md
CHANGED
|
@@ -24,7 +24,7 @@ from transformers import BertTokenizer, GPT2LMHeadModel, TextGenerationPipeline
|
|
| 24 |
|
| 25 |
fact_description = "1、2013年6月25日9时许，被告人丁某某在平阴县中医院建筑工地工人宿舍，窃取被害人胡某（男，43岁）现金1500元，在逃离现场时被工地工人抓获，丁某某将窃取的现金返还被害人。2、2013年7月12日14时许，被告人丁某某在平阴县府前街文鼎嘉苑建筑工地工人宿舍，窃取被害人陈某（男，31岁）及王某（男，25岁）现金850元，在逃跑时被抓获，丁某某将盗窃现金返还被害人。本院认为，"
|
| 26 |
|
| 27 |
-
model_name = "
|
| 28 |
|
| 29 |
device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
|
| 30 |
tokenizer = BertTokenizer.from_pretrained(model_name)
|
|
|
|
| 24 |
|
| 25 |
fact_description = "1、2013年6月25日9时许，被告人丁某某在平阴县中医院建筑工地工人宿舍，窃取被害人胡某（男，43岁）现金1500元，在逃离现场时被工地工人抓获，丁某某将窃取的现金返还被害人。2、2013年7月12日14时许，被告人丁某某在平阴县府前街文鼎嘉苑建筑工地工人宿舍，窃取被害人陈某（男，31岁）及王某（男，25岁）现金850元，在逃跑时被抓获，丁某某将盗窃现金返还被害人。本院认为，"
|
| 26 |
|
| 27 |
+
model_name = "seudl/aijudge"
|
| 28 |
|
| 29 |
device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
|
| 30 |
tokenizer = BertTokenizer.from_pretrained(model_name)
|