sca255 commited on
Commit
97915e4
·
verified ·
1 Parent(s): 622ec0d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -1
app.py CHANGED
@@ -2,11 +2,14 @@ import gradio as gr
2
  import hf_transfer
3
  from transformers import AutoModelForCausalLM, AutoTokenizer,StoppingCriteriaList,TextIteratorStreamer
4
  from threading import Thread
 
 
5
 
6
  model = AutoModelForCausalLM.from_pretrained(
7
  "mistralai/Mistral-7B-v0.1",
 
8
  )
9
- tknz=AutoTokenizer.from_pretrained("kubernetes-bad/chargen-v2")
10
 
11
 
12
 
 
2
  import hf_transfer
3
  from transformers import AutoModelForCausalLM, AutoTokenizer,StoppingCriteriaList,TextIteratorStreamer
4
  from threading import Thread
5
+ import os
6
+ HFTOKEN=os.getenv("hftoken")
7
 
8
  model = AutoModelForCausalLM.from_pretrained(
9
  "mistralai/Mistral-7B-v0.1",
10
+ token = HFTOKEN
11
  )
12
+ tknz=AutoTokenizer.from_pretrained("kubernetes-bad/chargen-v2",token=HFTOKEN)
13
 
14
 
15