fix on finBERT call

This commit is contained in:
Simon Moisy committed 2025-03-22 04:59:01 +08:00
1 parent 8784e29cb2
commit 247c59b800
1 file changed
+1 -1
+1 -1
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