Update inference.py

inference.py

@@ -16,8 +16,6 @@ model.to(device)
 
 import json
 
-# Mappa gli ID predetti nelle etichette di classificazione
-labels = processor.tokenizer.convert_ids_to_tokens(predicted_ids)
 
 # Funzione per creare l'output JSON in formato CORD-like
 def create_json_output(words, labels, boxes):
@@ -50,6 +48,9 @@ def prediction(image):
     logits = outputs.logits
     predicted_ids = logits.argmax(-1).squeeze().tolist()
     
+    # Mappa gli ID predetti nelle etichette di classificazione
+    labels = processor.tokenizer.convert_ids_to_tokens(predicted_ids)
+    
     predictions = outputs.logits.argmax(-1).squeeze().tolist()
     token_boxes = encoding.bbox.squeeze().tolist()
     probabilities = torch.softmax(outputs.logits, dim=-1)