jflo commited on
Commit
58fc7ed
·
1 Parent(s): e90a0ff

Added os getenv for key

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -36,7 +36,7 @@ class MultiHeadBERT(nn.Module):
36
  super(MultiHeadBERT, self).__init__()
37
 
38
  # Shared BERT backbone
39
- self.bert = BertModel.from_pretrained('bert-base-uncased',token=os.get_env('HF_TOKEN'))
40
  hidden_size = self.bert.config.hidden_size # 768
41
 
42
  # Task-specific classification heads
 
36
  super(MultiHeadBERT, self).__init__()
37
 
38
  # Shared BERT backbone
39
+ self.bert = BertModel.from_pretrained('bert-base-uncased',token=os.getenv('HF_TOKEN'))
40
  hidden_size = self.bert.config.hidden_size # 768
41
 
42
  # Task-specific classification heads