|
|
from transformers import pipeline |
|
|
from PIL import Image |
|
|
|
|
|
|
|
|
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.") |
|
|
|
|
|
|
|
|
|
|
|
|
