root commited on
Commit ·
7ca5bb0
1
Parent(s): 697fd0c
Ajout du code de prédiction
Browse files- 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()
|