Update app.py
Browse files
app.py
CHANGED
@@ -10,8 +10,8 @@ from transformers import AutoModelForCausalLM, AutoTokenizer
|
|
10 |
|
11 |
# Constants
|
12 |
# =========
|
13 |
-
MODEL_NAME = "
|
14 |
-
SYSTEM_MESSAGE = "
|
15 |
|
16 |
# Load Model and Tokenizer
|
17 |
# ========================
|
|
|
10 |
|
11 |
# Constants
|
12 |
# =========
|
13 |
+
MODEL_NAME = "prithivMLmods/Llama-Magpie-3.2-3B-Instruct"
|
14 |
+
SYSTEM_MESSAGE = "you are an AI assistant, and your name is Llama-Magpie-3.2-3B-Instruct"
|
15 |
|
16 |
# Load Model and Tokenizer
|
17 |
# ========================
|