radussad commited on
Commit
a2ca040
·
verified ·
1 Parent(s): 71e9b01

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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/phi3-mini-4k-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,
 
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,