songs1 commited on
Commit
7aa26ac
·
1 Parent(s): caf6fcf

pickup token from env var

Browse files
Files changed (1) hide show
  1. app.py +11 -2
app.py CHANGED
@@ -1,11 +1,20 @@
 
 
1
  import gradio as gr
2
  import spaces
3
  from guidance import gen
4
  from guidance.models import Transformers
5
  from transformers import AutoModelForCausalLM, AutoTokenizer
6
 
7
- model = AutoModelForCausalLM.from_pretrained("meta-llama/Llama-3.2-1B-Instruct")
8
- tok = AutoTokenizer.from_pretrained("meta-llama/Llama-3.2-1B-Instruct")
 
 
 
 
 
 
 
9
  model = model.to("cuda").eval()
10
 
11
 
 
1
+ import os
2
+
3
  import gradio as gr
4
  import spaces
5
  from guidance import gen
6
  from guidance.models import Transformers
7
  from transformers import AutoModelForCausalLM, AutoTokenizer
8
 
9
+ AUTH_TOKEN = os.environ.get("HF_TOKEN", False)
10
+ model = AutoModelForCausalLM.from_pretrained(
11
+ "meta-llama/Llama-3.2-1B-Instruct",
12
+ token=AUTH_TOKEN,
13
+ )
14
+ tok = AutoTokenizer.from_pretrained(
15
+ "meta-llama/Llama-3.2-1B-Instruct",
16
+ token=AUTH_TOKEN,
17
+ )
18
  model = model.to("cuda").eval()
19
 
20