noelstan99 commited on
Commit
f2a47fd
·
1 Parent(s): f2a1738

phi model update

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -12,7 +12,7 @@ st.set_page_config(
12
  # Load model and tokenizer
13
  @st.cache_resource
14
  def load_model():
15
- model_name = "meta-llama/Llama-3.1-8B" # Replace with the model path or identifier
16
  tokenizer = AutoTokenizer.from_pretrained(model_name)
17
  model = AutoModelForCausalLM.from_pretrained(model_name, torch_dtype=torch.float32)
18
  return tokenizer, model
 
12
  # Load model and tokenizer
13
  @st.cache_resource
14
  def load_model():
15
+ model_name = "ML-GT/CS4641-7641-finetuned-phi-3-mini-128k-instruct" # Replace with the model path or identifier
16
  tokenizer = AutoTokenizer.from_pretrained(model_name)
17
  model = AutoModelForCausalLM.from_pretrained(model_name, torch_dtype=torch.float32)
18
  return tokenizer, model