From b4ef1ad8a2c724529e90b7b136d53c276320197b Mon Sep 17 00:00:00 2001
From: Simon Moisy <simon.moisy@tutanota.com>
Date: Sat, 22 Mar 2025 05:13:12 +0800
Subject: [PATCH] updated model call

---
 article_analyzer.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

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