valentin urena commited on
Commit
74cda9e
·
verified ·
1 Parent(s): 8b41b73

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -22,9 +22,9 @@ 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
- # model_id = 'kaggle://valentinbaltazar/gemma-chess/keras/gemma_2b_en_chess'
28
 
29
 
30
  model = keras_nlp.models.GemmaCausalLM.from_preset(model_id)
 
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
+ model_id = 'kaggle://valentinbaltazar/gemma-chess/keras/gemma_2b_en_chess'
28
 
29
 
30
  model = keras_nlp.models.GemmaCausalLM.from_preset(model_id)