tigeryfan commited on
Commit
d3a24f2
·
verified ·
1 Parent(s): b363a98

Update backend.py

Browse files
Files changed (1) hide show
  1. backend.py +5 -0
backend.py CHANGED
@@ -1,5 +1,10 @@
1
  from transformers import pipeline
2
  from transformers import AutoTokenizer, AutoModelForCausalLM, pipeline
 
 
 
 
 
3
 
4
  def load_model():
5
  tokenizer = AutoTokenizer.from_pretrained("google/gemma-3-270m-it")
 
1
  from transformers import pipeline
2
  from transformers import AutoTokenizer, AutoModelForCausalLM, pipeline
3
+ import os
4
+ from huggingface_hub import login
5
+
6
+ access_token=os.getenv('hf_gated_model_access_token')
7
+ login(token=access_token)
8
 
9
  def load_model():
10
  tokenizer = AutoTokenizer.from_pretrained("google/gemma-3-270m-it")