kazuma313 commited on
Commit
1265201
1 Parent(s): 556d27c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -7,6 +7,7 @@ import re
7
  model_name = "google/gemma-2b"
8
  peft_model = "kazuma313/gemma-dokter-ft"
9
  device_map = "auto"
 
10
 
11
  base_model = AutoModelForCausalLM.from_pretrained(
12
  model_name,
 
7
  model_name = "google/gemma-2b"
8
  peft_model = "kazuma313/gemma-dokter-ft"
9
  device_map = "auto"
10
+ hf_token = userdata.get('hftoken')
11
 
12
  base_model = AutoModelForCausalLM.from_pretrained(
13
  model_name,