Rafii commited on
Commit
d920053
·
1 Parent(s): 157119a

should work

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -27,7 +27,8 @@ if st.button("Submit"):
27
  # Tokenize input and make predictions
28
  # inputs = tokenizer(user_input, return_tensors="pt")
29
  # outputs = model(**inputs)
30
- model, tokenizer = load("Rafii/f1llama")
 
31
 
32
  prompt="hello"
33
 
 
27
  # Tokenize input and make predictions
28
  # inputs = tokenizer(user_input, return_tensors="pt")
29
  # outputs = model(**inputs)
30
+ # model, tokenizer = load("Rafii/f1llama")
31
+ model, tokenizer = load("mlx-community/Meta-Llama-3.1-8B-Instruct-bf16")
32
 
33
  prompt="hello"
34