File size: 2,489 Bytes
86973d1 260705a 6ec80ce 86973d1 151c00f 6ec80ce 86973d1 6838fd4 86973d1 b0008b9 94b8fbb 975ddf4 fd216f3 975ddf4 fd216f3 6ec80ce 975ddf4 fd216f3 8a2d0b7 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 |
from transformers import pipeline
from PIL import Image
# Charger le modèle de classification
model = pipeline("image-classification", model="facebook/deit-base-distilled-patch16-224", device=0)
def predict(image: Image.Image):
image = image.convert("RGB").resize((224, 224))
predictions = model(image)
label = predictions[0]['label']
confidence = predictions[0]['score']
return label, confidence
def get_advice(label: str) -> str:
print(f"Label détecté : {label}")
advice_dict = {
"Golden Retriever": "Tempérament : Doux, amical, sociable\nTaille moyenne : 55 à 60 cm\nActivité : Élevé, aime courir et nager\nEspérance de vie : 10 à 12 ans",
"Labrador Retriever": "Tempérament : Amical, intelligent, joueur\nTaille moyenne : 55 à 62 cm\nActivité : Très élevé, adore les longues balades\nEspérance de vie : 12 à 14 ans",
"German Shepherd": "Tempérament : Loyal, protecteur, courageux\nTaille moyenne : 60 à 65 cm\nActivité : Très actif, besoin d'entraînement régulier\nEspérance de vie : 9 à 13 ans",
"Bulldog": "Tempérament : Calme, affectueux, têtu\nTaille moyenne : 35 à 40 cm\nActivité : Faible, préfère les petites promenades\nEspérance de vie : 8 à 10 ans",
"Beagle": "Tempérament : Curieux, joyeux, gentil\nTaille moyenne : 33 à 41 cm\nActivité : Élevé, adore explorer\nEspérance de vie : 12 à 15 ans",
"Eskimo dog, husky": "Tempérament : Indépendant, énergique, loyal\nTaille moyenne : 50 à 60 cm\nActivité : Très élevé, aime courir et tirer des traîneaux\nEspérance de vie : 12 à 15 ans",
"Maltese dog, Maltese terrier, Maltese": "Tempérament : Joyeux, sociable, affectueux\nTaille moyenne : 23 à 30 cm\nActivité : Modéré, aime jouer et se promener\nEspérance de vie : 14 à 16 ans",
"Bichon Frisé": "Tempérament : Joyeux, sociable, affectueux\nTaille moyenne : 23 à 30 cm\nActivité : Modéré, aime jouer et se promener\nEspérance de vie : 14 à 16 ans",
"bull mastiff": "Tempérament : Calme, courageux, fidèle\nTaille moyenne : 61 à 68 cm\nActivité : Moyenne, aime les courtes promenades\nEspérance de vie : 8 à 10 ans",
"basenji": "Tempérament : Indépendant, alerte, affectueux\nTaille moyenne : 40 à 43 cm\nActivité : Élevée, aime courir et jouer\nEspérance de vie : 12 à 16 ans"
}
return advice_dict.get(label, "Aucune information disponible pour cette race.")
|