Update app.py
Browse files
app.py
CHANGED
@@ -4,7 +4,7 @@ from transformers import AutoModelForCausalLM, AutoTokenizer
|
|
4 |
# Load the model and tokenizer
|
5 |
@st.cache_resource
|
6 |
def load_model_and_tokenizer():
|
7 |
-
model_name_or_path = "
|
8 |
model = AutoModelForCausalLM.from_pretrained(model_name_or_path, device_map="auto")
|
9 |
tokenizer = AutoTokenizer.from_pretrained(model_name_or_path)
|
10 |
return model, tokenizer
|
|
|
4 |
# Load the model and tokenizer
|
5 |
@st.cache_resource
|
6 |
def load_model_and_tokenizer():
|
7 |
+
model_name_or_path = "mistralai/Mistral-7B-Instruct-v0.2"
|
8 |
model = AutoModelForCausalLM.from_pretrained(model_name_or_path, device_map="auto")
|
9 |
tokenizer = AutoTokenizer.from_pretrained(model_name_or_path)
|
10 |
return model, tokenizer
|