Spaces:

GenAIDevTOProd
/

Reddit-SemanticSearch-Prototype

Sleeping

GenAIDevTOProd commited on Aug 6

Commit

cc97202

verified ·

1 Parent(s): 7afaf6d

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -19,7 +19,7 @@ HF_TOKEN = os.environ.get("RedditSemanticSearch")
 from datasets import load_dataset
 # Load full Reddit dataset (assumes it's pre-split by subreddit or has a field)
-dataset = load_dataset("HuggingFaceGECLM/REDDIT_comments", split="train")
 # Define target subreddits
 target_subreddits = ["askscience", "gaming", "technology", "todayilearned", "programming"]

 from datasets import load_dataset
 # Load full Reddit dataset (assumes it's pre-split by subreddit or has a field)
+dataset = load_dataset("HuggingFaceGECLM/REDDIT_comments", split="askscience", "gaming", "technology", "todayilearned", "programming")
 # Define target subreddits
 target_subreddits = ["askscience", "gaming", "technology", "todayilearned", "programming"]