Update inference.py

inference.py

@@ -24,16 +24,8 @@ def prediction(image):
     offset_mapping = encoding.pop('offset_mapping')
 
     # Esegui l'inferenza con il modello fine-tuned
-    with torch.no_grad():
-        outputs = model(**encoding)
-    
-    # Ottieni le etichette previste dal modello
-    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)
-    
+    outputs = model(**encoding)
+
     predictions = outputs.logits.argmax(-1).squeeze().tolist()
     token_boxes = encoding.bbox.squeeze().tolist()
     probabilities = torch.softmax(outputs.logits, dim=-1)
@@ -77,7 +69,7 @@ def prediction(image):
     draw = ImageDraw.Draw(image, "RGBA")
     font = ImageFont.load_default()
     
-    for prediction, box, confidence in zip(predictions, token_boxes, confidence_scores):
+    for prediction, box, confidence in zip(true_predictions, token_boxes, confidence_scores):
             draw.rectangle(box)
             draw.text((box[0]+10, box[1]-10), text=str(prediction)+ ", "+ str(confidence), font=font, fill="black", font_size="15")