valentin urena commited on
Commit
817e88e
·
verified ·
1 Parent(s): a8472ea

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -21,7 +21,9 @@ MAX_INPUT_TOKEN_LENGTH = 4096
21
  MAX_NEW_TOKENS = 2048
22
  DEFAULT_MAX_NEW_TOKENS = 128
23
 
24
- model_id = "hf://google/gemma-2b-keras"
 
 
25
 
26
 
27
  model = keras_nlp.models.GemmaCausalLM.from_preset(model_id)
 
21
  MAX_NEW_TOKENS = 2048
22
  DEFAULT_MAX_NEW_TOKENS = 128
23
 
24
+ # model_id = "hf://google/gemma-2b-keras"
25
+ model_id = "hf://google/gemma-2-2b-it"
26
+
27
 
28
 
29
  model = keras_nlp.models.GemmaCausalLM.from_preset(model_id)