This commit is contained in:
Simon Moisy committed 2025-03-22 04:47:52 +08:00
1 parent 4f665db568
commit 885f51b83d
1 file changed
+9 -2
+9 -2
View File
@@ -74,5 +74,12 @@ class ArticleAnalyzer:
def classify_article_finbert(self, article_html): def classify_article_finbert(self, article_html):
article_md = self.convert_to_markdown(article_html) article_md = self.convert_to_markdown(article_html)
result = self.classifier(article_md) chunk_size = 512
return result chunks = [article_md[i:i + chunk_size] for i in range(0, len(article_md), chunk_size)]
results = []
for chunk in chunks:
result = self.classifier(chunk)
results.append(result)
return results