Simon/CryptoMarketParser
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

updated model call

Browse Source
This commit is contained in:
Simon Moisy 2025-03-22 05:13:12 +08:00
parent c156898316
commit b4ef1ad8a2
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
article_analyzer.py
View File

@ -25,7 +25,7 @@ class ArticleClassification(BaseModel):
class ArticleAnalyzer: class ArticleAnalyzer:
def __init__(self): 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 = """ self.base_prompt = """
Classify the following article into one of these categories: Classify the following article into one of these categories:
- Regulatory News - Regulatory News