Update README.md
Browse files
README.md
CHANGED
@@ -37,9 +37,9 @@ device = "cuda" # the device to load the model onto
|
|
37 |
model = AutoModelForCausalLM.from_pretrained("mistralai/Mistral-7B-Instruct-v0.1")
|
38 |
tokenizer = AutoTokenizer.from_pretrained("mistralai/Mistral-7B-Instruct-v0.1")
|
39 |
|
40 |
-
text = "<s>[INST]
|
41 |
-
"
|
42 |
-
"[INST]
|
43 |
|
44 |
encodeds = tokenizer(text, return_tensors="pt", add_special_tokens=False)
|
45 |
|
|
|
37 |
model = AutoModelForCausalLM.from_pretrained("mistralai/Mistral-7B-Instruct-v0.1")
|
38 |
tokenizer = AutoTokenizer.from_pretrained("mistralai/Mistral-7B-Instruct-v0.1")
|
39 |
|
40 |
+
text = "<s>[INST] Below is a conversation transcript [/INST]"
|
41 |
+
"Your credit card has been stolen, and you need to contact us to resolve the issue. We will help you protect your information and prevent further fraud.</s> "
|
42 |
+
"[INST] Analyze the conversation and determine if it's fraudulent or legitimate. [/INST]"
|
43 |
|
44 |
encodeds = tokenizer(text, return_tensors="pt", add_special_tokens=False)
|
45 |
|