Dragneel commited on
Commit
301d19a
·
1 Parent(s): c9d8086

Update sentiment.py

Browse files
Files changed (1) hide show
  1. sentiment.py +2 -2
sentiment.py CHANGED
@@ -53,7 +53,7 @@ model.classifier = torch.nn.Linear(768, 2)
53
 
54
 
55
  def get_sentiment(query):
56
-
57
  filename = f"data/sentiment_analysis/{query}_results.pkl"
58
  saved_data = load_data(filename)
59
 
@@ -81,7 +81,7 @@ def get_reddit_results(query):
81
 
82
  try:
83
  sub = reddit.subreddit('noveltranslations+progressionfantasy')
84
- results = sub.search(query, limit=2)
85
 
86
 
87
  results_list = list(results)
 
53
 
54
 
55
  def get_sentiment(query):
56
+ query = query.lower()
57
  filename = f"data/sentiment_analysis/{query}_results.pkl"
58
  saved_data = load_data(filename)
59
 
 
81
 
82
  try:
83
  sub = reddit.subreddit('noveltranslations+progressionfantasy')
84
+ results = sub.search(query, limit=1)
85
 
86
 
87
  results_list = list(results)