GenAIDevTOProd commited on
Commit
39343bf
·
verified ·
1 Parent(s): 5725b7b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -5
app.py CHANGED
@@ -10,15 +10,17 @@ import faiss
10
  import gradio as gr
11
  from sklearn.metrics.pairwise import cosine_similarity
12
  from huggingface_hub import hf_hub_download, login
13
- from dotenv import load_dotenv
14
  from pyspark.sql import SparkSession
15
  from pyspark.sql.functions import col, udf, monotonically_increasing_id, collect_list, concat_ws
16
  from pyspark.sql.types import StringType
 
17
 
18
- # Load environment variables
19
- load_dotenv()
20
- HF_TOKEN = os.getenv("HF_TOKEN")
21
- login(token=HF_TOKEN)
 
 
22
 
23
  # Define target subreddits
24
  target_subreddits = ["askscience", "gaming", "technology", "todayilearned", "programming"]
 
10
  import gradio as gr
11
  from sklearn.metrics.pairwise import cosine_similarity
12
  from huggingface_hub import hf_hub_download, login
 
13
  from pyspark.sql import SparkSession
14
  from pyspark.sql.functions import col, udf, monotonically_increasing_id, collect_list, concat_ws
15
  from pyspark.sql.types import StringType
16
+ from huggingface_hub import HfApi
17
 
18
+ # Load token from Hugging Face Secrets (not .env)
19
+ HF_TOKEN = os.environ.get("RedditSemanticSearch")
20
+
21
+ # Authenticate
22
+ api = HfApi()
23
+ api.set_access_token(HF_TOKEN)
24
 
25
  # Define target subreddits
26
  target_subreddits = ["askscience", "gaming", "technology", "todayilearned", "programming"]