Update app.py

app.py

@@ -6,8 +6,10 @@ import gradio as gr
 import whisper
 import pathlib
 
+auth_token = os.environ.get("auth_key")
+
 #Get the netflix dataset
-netflix = load_dataset('hugginglearners/netflix-shows',use_auth_token=True)
+netflix = load_dataset('hugginglearners/netflix-shows',use_auth_token=auth_token)
 
 #load ASR model
 asr_model = whisper.load_model("small")
@@ -21,7 +23,7 @@ passages = netflix_df['description'].tolist()
 model = SentenceTransformer('all-mpnet-base-v2')
 
 #load embeddings
-flix_ds = load_dataset("nickmuchi/netflix-shows-mpnet-embeddings", use_auth_token=True)
+flix_ds = load_dataset("nickmuchi/netflix-shows-mpnet-embeddings", use_auth_token=auth_token)
 dataset_embeddings = torch.from_numpy(flix_ds["train"].to_pandas().to_numpy()).to(torch.float)
 
 #load cross-encoder for reranking