OmPrakashSingh1704 commited on
Commit
58d8835
1 Parent(s): 98f6a5e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -9,12 +9,12 @@ login(token=os.getenv("TOKEN"))
9
 
10
  # Load model and tokenizer
11
  model = AutoModelForCausalLM.from_pretrained(
12
- "mistralai/Mistral-7B-Instruct-v0.2",
13
  torch_dtype="auto",
14
  device_map="auto",
15
  )
16
 
17
- tokenizer = AutoTokenizer.from_pretrained("mistralai/Mistral-7B-Instruct-v0.2")
18
  if 'recipe' not in st.session_state:
19
  st.session_state.recipe = None
20
 
 
9
 
10
  # Load model and tokenizer
11
  model = AutoModelForCausalLM.from_pretrained(
12
+ "google/gemma-7b-it",
13
  torch_dtype="auto",
14
  device_map="auto",
15
  )
16
 
17
+ tokenizer = AutoTokenizer.from_pretrained("google/gemma-7b-it")
18
  if 'recipe' not in st.session_state:
19
  st.session_state.recipe = None
20