kazuma313 commited on
Commit
dd3757f
1 Parent(s): 1265201

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -3,11 +3,12 @@ from peft import PeftModel, PeftConfig
3
  from transformers import AutoModelForCausalLM, AutoTokenizer
4
  import torch
5
  import re
 
6
 
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,
 
3
  from transformers import AutoModelForCausalLM, AutoTokenizer
4
  import torch
5
  import re
6
+ import os
7
 
8
  model_name = "google/gemma-2b"
9
  peft_model = "kazuma313/gemma-dokter-ft"
10
  device_map = "auto"
11
+ hf_token = os.getenv('hftoken')
12
 
13
  base_model = AutoModelForCausalLM.from_pretrained(
14
  model_name,