ai-bookkeeper-backend / test_classifier.py
Pushkar Pandey
fix: move backend files to root for HuggingFace
61ec498
Raw
History Blame Contribute Delete
291 Bytes
from transformers import pipeline
try:
print("Loading pipeline...")
classifier = pipeline("text-classification", model="finmigodeveloper/distilbert-transaction-classifier-lora")
res = classifier("Starbucks coffee")
print(res)
except Exception as e:
print(f"Error: {e}")