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 finBERT.finbert import predict
from finBERT.finbert.finbert import predict
from transformers import AutoModelForSequenceClassification
import ollama
from pydantic import BaseModel