Salvando alterações

app.py

@@ -1,4 +1,4 @@
-from transformers import AutoProcessor, AutoModelForVision2Seq
+from transformers import AutoProcessor, AutoModel
 from PIL import Image
 import torch
 
@@ -12,12 +12,12 @@ prompt = "Extraia os seguintes dados do documento: [nome, cpf e data de nascimen
 device = torch.device("cpu")
 torch_dtype = torch.float32  # Usar precisão simples para economizar memória
 
-# Adicione trust_remote_code=True aqui 👇
 processor = AutoProcessor.from_pretrained(model_id, trust_remote_code=True)
-model = AutoModelForVision2Seq.from_pretrained(
+
+model = AutoModel.from_pretrained(
     model_id,
     torch_dtype=torch_dtype,
-    trust_remote_code=True  # E aqui 👇
+    trust_remote_code=True
 ).to(device)
 
 # Preparar inputs