diff --git a/article_analyzer.py b/article_analyzer.py index 9c9da6f..103bbf3 100644 --- a/article_analyzer.py +++ b/article_analyzer.py @@ -25,7 +25,7 @@ class ArticleClassification(BaseModel): class ArticleAnalyzer: def __init__(self): - self.model = AutoModelForSequenceClassification.from_pretrained('args.model_path', num_labels=3, cache_dir=None) + self.model = AutoModelForSequenceClassification.from_pretrained("ProsusAI/finbert", num_labels=3, cache_dir=None) self.base_prompt = """ Classify the following article into one of these categories: - Regulatory News