fix on finBERT call

This commit is contained in:
Simon Moisy 2025-03-22 04:59:01 +08:00
parent 8784e29cb2
commit 247c59b800

View File

@ -1,5 +1,5 @@
from enum import Enum from enum import Enum
from finBERT.finbert import predict from finBERT.finbert.finbert import predict
from transformers import AutoModelForSequenceClassification from transformers import AutoModelForSequenceClassification
import ollama import ollama
from pydantic import BaseModel from pydantic import BaseModel