Update app.py

app.py

@@ -17,6 +17,7 @@ from collections import defaultdict
 from asyncpraw.models import MoreComments, Submission
 from tqdm import tqdm
 from huggingface_hub import InferenceClient, notebook_login
+from transformers import pipeline
 from datetime import datetime, timedelta
 from datasets import load_dataset
 from helper import get_access_to_reddit, get_write_access_to_hf, search_subreddits_by_keyword_in_name_or_description, filter_subreddits_by_keywords, get_subreddits_name_title_description, process_output, probe_subs_for_posts, default_dict_dict_dict_list, probe_submissions_for_comments, results_str_to_dict
@@ -68,7 +69,8 @@ async def main_async():
   # -- pass new subreddits through classifier to determine if they are technology related --
   print("passing new subreddits through classifier to determine if they are technology related...")
 
-  topic_classifier_client = InferenceClient(model="gulnuravci/subreddit_description_topic_classifier", token=os.getenv("REDDIT_READ"))
+  # topic_classifier_client = InferenceClient(model="gulnuravci/subreddit_description_topic_classifier", token=os.getenv("REDDIT_READ"))
+  topic_classifier_client = pipeline("text-classification", model="gulnuravci/subreddit_description_topic_classifier")
 
   # key is the parent company and the value is a list of subreddit objects that are technology related
   subreddits_passed_topic_classifier = defaultdict(list)