kehanlu commited on
Commit
37ce538
·
1 Parent(s): f0a4357
Files changed (1) hide show
  1. app.py +1 -3
app.py CHANGED
@@ -7,11 +7,9 @@ import gradio as gr
7
  from transformers import AutoModel
8
  import os
9
 
10
- token = os.getenv('token')
11
-
12
 
13
  if gr.NO_RELOAD:
14
- model = AutoModel.from_pretrained("DeSTA-ntu/DeSTA2-8B-beta", trust_remote_code=True, token=token)
15
  model.to("cuda")
16
  model.eval()
17
 
 
7
  from transformers import AutoModel
8
  import os
9
 
 
 
10
 
11
  if gr.NO_RELOAD:
12
+ model = AutoModel.from_pretrained("DeSTA-ntu/DeSTA2-8B-beta", trust_remote_code=True, token=os.getenv('HF_TOKEN'))
13
  model.to("cuda")
14
  model.eval()
15