mirror of
https://github.com/Ladebeze66/llm_ticket3.git
synced 2025-12-13 09:06:51 +01:00
29 lines
938 B
Python
29 lines
938 B
Python
from agents.llama_vision.agent_vision_ocr import AgentVisionOCR
|
|
from llm_classes.llama_vision import LlamaVision
|
|
import json
|
|
import os
|
|
|
|
# Créer les répertoires de résultats s'ils n'existent pas
|
|
os.makedirs("results/ocr_brut", exist_ok=True)
|
|
os.makedirs("results/ocr_avance", exist_ok=True)
|
|
|
|
# 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")
|
|
|
|
# Affichage des chemins de sortie
|
|
print(f"✅ Résultats OCR brut sauvegardés dans: results/ocr_brut/")
|
|
print(f"✅ Résultats OCR avancé sauvegardés dans: results/ocr_avance/")
|