updated model call

2025-03-22 05:13:12 +08:00 · 2025-03-22 05:13:12 +08:00 · b4ef1ad8a2
commit b4ef1ad8a2
parent c156898316
1 changed files with 1 additions and 1 deletions
--- 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