Update sentiment.py
Browse files- 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=
|
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)
|