Update README.md
Browse files
README.md
CHANGED
@@ -22,8 +22,8 @@ from transformers import AutoTokenizer, AutoModelForCausalLM
|
|
22 |
import re
|
23 |
|
24 |
# Load model and tokenizer
|
25 |
-
tokenizer = AutoTokenizer.from_pretrained("reglab/mistral-rrc")
|
26 |
-
model = AutoModelForCausalLM.from_pretrained("reglab/mistral-rrc")
|
27 |
|
28 |
def format_prompt(document):
|
29 |
return f"""### Instruction:
|
|
|
22 |
import re
|
23 |
|
24 |
# Load model and tokenizer
|
25 |
+
tokenizer = AutoTokenizer.from_pretrained("reglab-rrc/mistral-rrc")
|
26 |
+
model = AutoModelForCausalLM.from_pretrained("reglab-rrc/mistral-rrc")
|
27 |
|
28 |
def format_prompt(document):
|
29 |
return f"""### Instruction:
|