From c156898316f9ae4510195a71ebf1bbcf7043ebfd Mon Sep 17 00:00:00 2001 From: Simon Moisy Date: Sat, 22 Mar 2025 05:03:54 +0800 Subject: [PATCH] added subpath --- main_article_analyzer.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/main_article_analyzer.py b/main_article_analyzer.py index 8f2eb6d..ed658a0 100644 --- a/main_article_analyzer.py +++ b/main_article_analyzer.py @@ -1,9 +1,12 @@ import os +import sys from sqlalchemy import create_engine, Column, String, Float, MetaData, Table from sqlalchemy.ext.declarative import declarative_base from sqlalchemy.orm import sessionmaker from article_analyzer import ArticleAnalyzer +sys.path.append(os.path.join(os.path.dirname(__file__), 'finBERT', 'finbert')) + Base = declarative_base() class ArticleAnalysis(Base): @@ -35,7 +38,7 @@ if __name__ == "__main__": print(f"Parsed {len(html_files)} html files") Base.metadata.create_all(engine) - + for file, content in html_files.items(): result = analyzer.classify_article_finbert(content)