tatihden commited on
Commit
ae0810b
·
verified ·
1 Parent(s): f248897

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -11,6 +11,8 @@ os.environ["XLA_PYTHON_CLIENT_MEM_FRACTION"]="1.00"
11
  import keras_hub
12
  import keras
13
 
 
 
14
  #quantization_config = BitsAndBytesConfig(load_in_8bit=True)
15
 
16
  models = [
 
11
  import keras_hub
12
  import keras
13
 
14
+ tokenizer = keras_hub.models.GemmaTokenizer.from_preset("hf://tatihden/gemma_mental_health_2b_it_en")
15
+
16
  #quantization_config = BitsAndBytesConfig(load_in_8bit=True)
17
 
18
  models = [