OmPrakashSingh1704 commited on
Commit
72e6cc9
1 Parent(s): 11d500b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -6,12 +6,12 @@ from datetime import datetime
6
 
7
  # Load model and tokenizer
8
  model = AutoModelForCausalLM.from_pretrained(
9
- "Qwen/Qwen1.5-0.5B-Chat",
10
  torch_dtype="auto",
11
  device_map="auto",
12
  )
13
 
14
- tokenizer = AutoTokenizer.from_pretrained("Qwen/Qwen1.5-0.5B-Chat")
15
  if 'recipe' not in st.session_state:
16
  st.session_state.recipe = None
17
 
 
6
 
7
  # Load model and tokenizer
8
  model = AutoModelForCausalLM.from_pretrained(
9
+ "mistralai/Mistral-7B-Instruct-v0.2",
10
  torch_dtype="auto",
11
  device_map="auto",
12
  )
13
 
14
+ tokenizer = AutoTokenizer.from_pretrained("mistralai/Mistral-7B-Instruct-v0.2")
15
  if 'recipe' not in st.session_state:
16
  st.session_state.recipe = None
17