Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -10,7 +10,7 @@ from retriever import retrieve_documents
|
|
10 |
|
11 |
# Load Mistral 7B model
|
12 |
#MODEL_NAME = "mistralai/Mistral-7B-v0.1"
|
13 |
-
MODEL_NAME = "microsoft/
|
14 |
|
15 |
tokenizer = AutoTokenizer.from_pretrained(MODEL_NAME, use_auth_token=os.getenv("HUGGING_FACE_HUB_TOKEN"), cache_dir="/tmp/huggingface")
|
16 |
model = AutoModelForCausalLM.from_pretrained(MODEL_NAME,
|
|
|
10 |
|
11 |
# Load Mistral 7B model
|
12 |
#MODEL_NAME = "mistralai/Mistral-7B-v0.1"
|
13 |
+
MODEL_NAME = "microsoft/Phi-4-mini-instruct"
|
14 |
|
15 |
tokenizer = AutoTokenizer.from_pretrained(MODEL_NAME, use_auth_token=os.getenv("HUGGING_FACE_HUB_TOKEN"), cache_dir="/tmp/huggingface")
|
16 |
model = AutoModelForCausalLM.from_pretrained(MODEL_NAME,
|