fix : using the right ckpts

app.py

@@ -1,10 +1,8 @@
 from flask import Flask, render_template,request, redirect,url_for, jsonify
-from PyPDF2 import PdfReader
 from helper_functions import predict_class
 import fitz  # PyMuPDF
 import os, shutil
 import torch
-from transformers import BertTokenizer, BertForSequenceClassification
 import pickle
 
 app = Flask(__name__)

helper_functions.py

@@ -1,15 +1,15 @@
 import torch
 import pickle
-from transformers import BertTokenizer, BertForSequenceClassification
+from transformers import  AutoTokenizer , DistilBertForSequenceClassification
 from transformers import BatchEncoding, PreTrainedTokenizerBase
 from typing import Optional
 from torch import Tensor
 
 # Load the model
-model = BertForSequenceClassification.from_pretrained("distilBert/DistilBert/saved_model")
+model = DistilBertForSequenceClassification.from_pretrained("distilBert/DistilBert/saved_model")
 
 # Load the tokenizer
-tokenizer = BertTokenizer.from_pretrained("distilBert/DistilBert/saved_tokenizer")
+tokenizer = AutoTokenizer.from_pretrained("distilBert/DistilBert/saved_tokenizer")
 
 # Charger le label encoder
 with open("distilBert/DistilBert/label_encoder.pkl", "rb") as f: