Add auth token

app.py

@@ -1,4 +1,5 @@
 import gradio as gr
+import os
 import torch
 import transformers
 
@@ -6,8 +7,8 @@ device = torch.device('cuda') if torch.cuda.is_available() else torch.device('cp
 
 class Interactive:
     def __init__(self):
-        self.tokenizer = transformers.AutoTokenizer.from_pretrained('liujch1998/cd-pi')
-        self.model = transformers.AutoModelForSeq2SeqLM.from_pretrained('liujch1998/cd-pi').to(device)
+        self.tokenizer = transformers.AutoTokenizer.from_pretrained('liujch1998/cd-pi', use_auth_token=os.environ['HF_TOKEN_DOWNLOAD'])
+        self.model = transformers.AutoModelForSeq2SeqLM.from_pretrained('liujch1998/cd-pi', use_auth_token=os.environ['HF_TOKEN_DOWNLOAD']).to(device)
         self.linear = torch.nn.Linear(self.model.shared.embedding_dim, 1).to(device)
         self.linear.weight = torch.nn.Parameter(self.model.shared.weight[32099, :].unsqueeze(0)) # (1, D)
         self.linear.bias = torch.nn.Parameter(self.model.shared.weight[32098, 0].unsqueeze(0)) # (1)