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)