Brian Morin
commited on
Commit
•
e1d2879
1
Parent(s):
e304ec8
Add application file
Browse files
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/
|
21 |
-
model = AutoModelForCausalLM.from_pretrained("brdemorin/
|
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:")
|