Brian Morin commited on
Commit
e1d2879
1 Parent(s): e304ec8

Add application file

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -17,8 +17,8 @@ x = st.slider('Select a value')
17
  st.write(x, 'squared is', x * x)
18
 
19
  # Load the tokenizer and model
20
- tokenizer = AutoTokenizer.from_pretrained("brdemorin/Phi3-custom_60-steps")
21
- model = AutoModelForCausalLM.from_pretrained("brdemorin/Phi3-custom_60-steps")
22
 
23
  # Create a text input for the user to enter their message
24
  user_input = st.text_input("Enter your message:")
 
17
  st.write(x, 'squared is', x * x)
18
 
19
  # Load the tokenizer and model
20
+ tokenizer = AutoTokenizer.from_pretrained("brdemorin/Phi3_80_steps")
21
+ model = AutoModelForCausalLM.from_pretrained("brdemorin/Phi3_80_steps")
22
 
23
  # Create a text input for the user to enter their message
24
  user_input = st.text_input("Enter your message:")