camphong24032002 commited on
Commit
dfd88e5
1 Parent(s): 339198c
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -35,7 +35,7 @@ def df_to_text(df):
35
  df = pd.read_csv("data/dataset.csv").reset_index(drop=True)
36
  embedding_df = pd.read_csv("data/embedding.csv", header=None)
37
  docs = embedding_df.values
38
- HF_TOKEN=os.environ("HF_TOKEN")
39
 
40
  model = SentenceTransformer("bert-base-nli-mean-tokens", cache_folder = "/code/", use_auth_token=HF_TOKEN)
41
 
 
35
  df = pd.read_csv("data/dataset.csv").reset_index(drop=True)
36
  embedding_df = pd.read_csv("data/embedding.csv", header=None)
37
  docs = embedding_df.values
38
+ HF_TOKEN=os.environ.get("HF_TOKEN")
39
 
40
  model = SentenceTransformer("bert-base-nli-mean-tokens", cache_folder = "/code/", use_auth_token=HF_TOKEN)
41