Update app.py

app.py

@@ -1,12 +1,12 @@
 import gradio as gr
-from transformers import AutoProcessor, AutoModelForDocumentClassification
+from transformers import AutoProcessor, LayoutLMForSequenceClassification
 from PIL import Image
 import torch
 
 # Charger le modèle et le processeur
 model_name = "impira/layoutlm-document-classifier"
 processor = AutoProcessor.from_pretrained(model_name)
-model = AutoModelForDocumentClassification.from_pretrained(model_name)
+model = LayoutLMForSequenceClassification.from_pretrained(model_name)
 
 def classify_document(image):
     # Préparer l'image pour le modèle
@@ -33,4 +33,4 @@ interface = gr.Interface(
 )
 
 # Lancer l'application
-interface.launch()
+interface.launch()