OmPrakashSingh1704 commited on
Commit
253a1e9
1 Parent(s): d460cfd

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -6,11 +6,11 @@ from datetime import datetime
6
 
7
  # Load model and tokenizer
8
  model = AutoModelForCausalLM.from_pretrained(
9
- "Qwen/Qwen2-72B-Instruct",
10
  torch_dtype="auto",
11
  device_map="auto",
12
  )
13
- tokenizer = AutoTokenizer.from_pretrained("Qwen/Qwen2-72B-Instruct")
14
 
15
  if 'recipe' not in st.session_state:
16
  st.session_state.recipe = None
 
6
 
7
  # Load model and tokenizer
8
  model = AutoModelForCausalLM.from_pretrained(
9
+ "meta-llama/Meta-Llama-3.1-8B-Instruct",
10
  torch_dtype="auto",
11
  device_map="auto",
12
  )
13
+ tokenizer = AutoTokenizer.from_pretrained("meta-llama/Meta-Llama-3.1-8B-Instruct")
14
 
15
  if 'recipe' not in st.session_state:
16
  st.session_state.recipe = None