root commited on
Commit
7ca5bb0
1 Parent(s): 697fd0c

Ajout du code de prédiction

Browse files
Files changed (1) hide show
  1. handler.py +12 -0
handler.py CHANGED
@@ -34,3 +34,15 @@ def predict(data):
34
 
35
  # Retourner le résultat au format JSON
36
  return json.dumps({"result": result})
 
 
 
 
 
 
 
 
 
 
 
 
 
34
 
35
  # Retourner le résultat au format JSON
36
  return json.dumps({"result": result})
37
+
38
+ # Fonction d'initialisation qui sera appelée lors du démarrage du service
39
+ init()
40
+
41
+ # Fonction de prédiction qui sera appelée pour traiter les requêtes d'inférence
42
+ def main():
43
+ data = {"inputs": "Votre texte d'entrée"} # Remplacez par les données d'entrée
44
+ result = predict(data)
45
+ print(result)
46
+
47
+ if __name__ == "__main__":
48
+ main()