updated model call

This commit is contained in:
Simon Moisy committed 2025-03-22 05:13:12 +08:00
1 parent c156898316
commit b4ef1ad8a2
1 file changed
+1 -1
+1 -1
View File
@@ -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