Update main.py

main.py

@@ -26,7 +26,7 @@ comment_dataset_name = f"{username}/reddit-comments-{subreddit}"
 dataset_readme_path = "README.md"
 
 frequency = os.environ.get("FREQUENCY", '').lower()
-if frequency not in ["daily", "hourly"]:
+if frequency not in ["daily", "hourly", "custom"]:
     raise ValueError("FREQUENCY environment variable must be 'daily' or 'hourly'")
 
 # Authenticate with Hugging Face using an auth token