Akshayram1 commited on
Commit
993065e
·
verified ·
1 Parent(s): 8098013

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -23,7 +23,7 @@ email_with_phone_regex = re.compile(r'(\d{10}).|.(\d{10})')
23
  # Load Phi-3 model and tokenizer
24
  @st.cache_resource
25
  def load_model():
26
- model_name = "microsoft/phi-3-mini-4k-instruct"
27
  tokenizer = AutoTokenizer.from_pretrained(model_name, trust_remote_code=True)
28
  model = AutoModel.from_pretrained(model_name, trust_remote_code=True)
29
  return tokenizer, model
 
23
  # Load Phi-3 model and tokenizer
24
  @st.cache_resource
25
  def load_model():
26
+ model_name = "microsoft/Phi-4-mini-instruct"
27
  tokenizer = AutoTokenizer.from_pretrained(model_name, trust_remote_code=True)
28
  model = AutoModel.from_pretrained(model_name, trust_remote_code=True)
29
  return tokenizer, model