mirror of
https://github.com/Ladebeze66/llm_ticket3.git
synced 2025-12-13 17:27:18 +01:00
25 lines
791 B
Python
25 lines
791 B
Python
from llm_classes.mistral_medium import MistralMedium
|
|
from llm_classes.pixtral_12b import Pixtral12b
|
|
|
|
print("Initialisation des modèles LLM...")
|
|
|
|
# Initialisation des modèles
|
|
try:
|
|
text_model = MistralMedium()
|
|
image_model = Pixtral12b()
|
|
print("Modèles initialisés avec succès!")
|
|
except Exception as e:
|
|
print(f"Erreur lors de l'initialisation des modèles: {str(e)}")
|
|
exit(1)
|
|
|
|
# Test d'interrogation simple
|
|
try:
|
|
question = "Quelle est la capitale de la France?"
|
|
print(f"\nTest d'interrogation simple sur MistralMedium:")
|
|
print(f"Question: {question}")
|
|
response = text_model.interroger(question)
|
|
print(f"Réponse: {response[:100]}...")
|
|
except Exception as e:
|
|
print(f"Erreur lors de l'interrogation simple: {str(e)}")
|
|
|
|
print("\nTests terminés!") |