Ajout du handler et des requirements pour l'API d'inférence

handler.py

@@ -87,4 +87,4 @@ class InferenceHandler:
         labels = {0: "Animaux", 1: "Capitales"}
         predicted_label = labels.get(prediction, "Inconnu")
         
-        return [{"label": predicted_label, "score": outputs.softmax(dim=1)[0][prediction].item()}]
+        return [{"label": predicted_label, "score": outputs.softmax(dim=1)[0][prediction].item()}]

requirements.txt

@@ -1 +1,2 @@
+
 torch