llm_ticket3/test_agent_ocr.py
2025-04-30 17:45:17 +02:00

20 lines
588 B
Python

from agents.llama_vision.agent_vision_ocr import AgentVisionOCR
from llm_classes.llama_vision import LlamaVision
import json
# Instanciation du modèle
model = LlamaVision()
agent = AgentVisionOCR(model)
# Affichage des paramètres actifs
print("🔵 Paramètres actifs LLM:")
print(json.dumps(agent.llm.params, indent=2))
# Image à tester
image_path = "output/ticket_T11143/T11143_20250422_084617/attachments/image_145435.png"
# Test de l'OCR
res = agent.executer(image_path)
print(f"\n🔵 Résultat pour l'image {image_path}:")
print(f"Texte extrait:\n{res['extracted_text']}\n")