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

to cuda

Browse Source
This commit is contained in:
Simon Moisy 2025-03-22 08:15:33 +08:00
parent 679f1bd941
commit 876f1f37a1
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
article_analyzer.py
View File

@ -26,6 +26,7 @@ class ArticleClassification(BaseModel):
class ArticleAnalyzer: class ArticleAnalyzer:
def __init__(self): def __init__(self):
self.model = AutoModelForSequenceClassification.from_pretrained("ProsusAI/finbert", num_labels=3, cache_dir=None) self.model = AutoModelForSequenceClassification.from_pretrained("ProsusAI/finbert", num_labels=3, cache_dir=None)
self.model.to("cuda")
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