OCR / app /application /services /document_classifier.py
Zeyadmohamed21's picture
Deploy with CPU-only fix for ZeroGPU compatibility
a39757f verified
Raw History Blame Contribute Delete
347 Bytes
from app.services.classification.classifier import UniversalDocumentClassifier
class DocumentClassifier:
"""
Backwards-compatible wrapper delegating to UniversalDocumentClassifier.
"""
@staticmethod
def classify(text: str) -> str:
doc_type, _ = UniversalDocumentClassifier.classify(text)
return doc_type