rohansampath commited on
Commit
4c36941
·
verified ·
1 Parent(s): dbd35bb

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -0
app.py CHANGED
@@ -8,6 +8,12 @@ matplotlib.use('Agg') # for non-interactive envs
8
  import matplotlib.pyplot as plt
9
  import io
10
  import base64
 
 
 
 
 
 
11
 
12
  # ---------------------------------------------------------------------------
13
  # 1. Define model name and load model/tokenizer
 
8
  import matplotlib.pyplot as plt
9
  import io
10
  import base64
11
+ import os
12
+ from huggingface_hub import login
13
+ from transformers import AutoTokenizer, AutoModel
14
+
15
+ hf_token = os.getenv("HF_TOKEN_READ_WRITE") # Read the token from Secrets
16
+ login(hf_token)
17
 
18
  # ---------------------------------------------------------------------------
19
  # 1. Define model name and load model/tokenizer