YsnHdn commited on
Commit
b6babbd
1 Parent(s): 871a3fb

Refactor : deleting pre-upload method

Browse files
Files changed (1) hide show
  1. app.py +0 -54
app.py CHANGED
@@ -17,60 +17,6 @@ def home():
17
  chart_data = dict()
18
  return render_template('pdf.html', class_probabilities= class_probabilities, predicted_class=predict_class,chart_data = chart_data)
19
 
20
- @app.route("/upload", methods=['POST'])
21
- def process_pdf():
22
- # Récupérer le fichier PDF de la requête
23
- file = request.files['file']
24
- filename = file.filename
25
-
26
- # Enregistrer le fichier dans le répertoire de téléchargement
27
- filepath = app.config['UPLOAD_FOLDER'] + "/" + filename
28
- file.save(filepath)
29
-
30
- # Ouvrir le fichier PDF
31
- pdf_document = fitz.open(filepath)
32
-
33
- # Initialiser une variable pour stocker le texte extrait
34
- extracted_text = ""
35
-
36
- # Boucler à travers chaque page pour extraire le texte
37
- for page_num in range(len(pdf_document)):
38
- # Récupérer l'objet de la page
39
- page = pdf_document.load_page(page_num)
40
-
41
- # Extraire le texte de la page
42
- page_text = page.get_text()
43
-
44
- # Ajouter le texte de la page à la variable d'extraction
45
- extracted_text += f"\nPage {page_num + 1}:\n{page_text}"
46
-
47
- # Fermer le fichier PDF
48
- pdf_document.close()
49
-
50
- # Charger le tokenizer
51
- tokenizer = BertTokenizer.from_pretrained(r"C:\Users\user\Desktop\3A\PFE\France\Stage PFE\models1-20240411T071206Z-001\models1\bert_tokenizer")
52
- # Charger le modèle
53
- model = BertForSequenceClassification.from_pretrained(r"C:\Users\user\Desktop\3A\PFE\France\Stage PFE\models1-20240411T071206Z-001\models1\bert_model")
54
-
55
- # Charger les labels
56
- with open(r"C:\Users\user\Desktop\3A\PFE\France\Stage PFE\models1-20240411T071206Z-001\models1\labels.pkl", 'rb') as f:
57
- labels = pickle.load(f)
58
-
59
- # Prétraiter le texte extrait
60
- inputs = tokenizer(extracted_text, return_tensors="pt", padding=True, truncation=True)
61
-
62
- # Passer l'entrée à travers le modèle pour obtenir les prédictions
63
- with torch.no_grad():
64
- outputs = model(**inputs)
65
-
66
- # Obtenir les prédictions de classe
67
- predicted_class_id = torch.argmax(outputs.logits, dim=1).item()
68
- predicted_class = labels[predicted_class_id]
69
-
70
- # Retourner la prédiction de classe ainsi que le texte extrait
71
- return render_template("home.html", extracted_text=extracted_text, predicted_class=predicted_class)
72
-
73
-
74
  @app.route('/pdf')
75
  def pdf():
76
  predict_class = ""
 
17
  chart_data = dict()
18
  return render_template('pdf.html', class_probabilities= class_probabilities, predicted_class=predict_class,chart_data = chart_data)
19
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
20
  @app.route('/pdf')
21
  def pdf():
22
  predict_class = ""