Update app.py

app.py

@@ -66,7 +66,7 @@ def process_image(image):
     with torch.no_grad():
         outputs = model(input_ids=input_ids, bbox=bbox, attention_mask=attention_mask)
 
-    predicted_labels = outputs.predictions.argmax(dim=2).squeeze().tolist()
+    predicted_labels = outputs.logits.argmax(dim=2).squeeze().tolist()
 
     # Extract content from boxes
     extracted_content = {}