diff --git a/src/model/pdf_classifier.py b/src/model/pdf_classifier.py index 6494452..21f4c69 100644 --- a/src/model/pdf_classifier.py +++ b/src/model/pdf_classifier.py @@ -20,6 +20,8 @@ def classify_pdf(pdf_path, model_dir="src/model/model-config"): model = joblib.load(model_path) vectorizer = joblib.load(vectorizer_path) + # Testing + # Extract text from PDF text = extract_text_from_pdf(pdf_path) if not text.strip():