ShubhamD95 commited on
Commit
f93c703
·
verified ·
1 Parent(s): c96b158

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +8 -0
app.py CHANGED
@@ -1,4 +1,12 @@
1
  from transformers import AutoTokenizer, AutoModelForCausalLM
 
 
 
 
 
 
 
 
2
 
3
  model_name = "google/gemma-3-1b-it"
4
  tokenizer = AutoTokenizer.from_pretrained(model_name)
 
1
  from transformers import AutoTokenizer, AutoModelForCausalLM
2
+ import os
3
+ from huggingface_hub import login
4
+
5
+
6
+ hf_token = os.environ.get("hf_space_token")
7
+
8
+
9
+ login(token=hf_token)
10
 
11
  model_name = "google/gemma-3-1b-it"
12
  tokenizer = AutoTokenizer.from_pretrained(model_name)