Update README.md
Browse files
README.md
CHANGED
|
@@ -21,7 +21,7 @@ model = AutoModelForCausalLM.from_pretrained(
|
|
| 21 |
torch_dtype="auto",
|
| 22 |
device_map="auto"
|
| 23 |
)
|
| 24 |
-
tokenizer = AutoTokenizer.from_pretrained("convoicon/
|
| 25 |
|
| 26 |
prompt = """You are a content classifier tasked with analyzing input content and determining if it falls within pre-defined metrics.
|
| 27 |
Don't forget to add reasoning in order to explain your classification result. Also do not add metrics which not included in the pre-defined metrics.
|
|
|
|
| 21 |
torch_dtype="auto",
|
| 22 |
device_map="auto"
|
| 23 |
)
|
| 24 |
+
tokenizer = AutoTokenizer.from_pretrained("convoicon/modestus_lightning")
|
| 25 |
|
| 26 |
prompt = """You are a content classifier tasked with analyzing input content and determining if it falls within pre-defined metrics.
|
| 27 |
Don't forget to add reasoning in order to explain your classification result. Also do not add metrics which not included in the pre-defined metrics.
|