File size: 2,480 Bytes
0666409 |
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 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 |
import base64
import os
import sys
# Ajoutez le chemin du dossier parent pour que Python puisse trouver handler.py
sys.path.insert(0, os.path.abspath(os.path.join(os.path.dirname(__file__))))
from handler import EndpointHandler # Assurez-vous que handler.py est dans le même dossier ou dans le PYTHONPATH
# --- Configuration pour le test local ---
# Le chemin vers le modèle n'est pas utilisé par notre handler car il charge le modèle par son ID
# Mais l'EndpointHandler attend un argument `path`.
# Pour le test local, on peut passer une chaîne vide ou un point.
MODEL_PATH_FOR_HANDLER = "."
# Chemin vers votre vidéo de test
# Assurez-vous que ce chemin est correct par rapport à l'endroit où vous exécutez ce script
VIDEO_FILE_PATH = "../../static/video-test.mp4" # Ajustez ce chemin si nécessaire
# --- Initialisation du handler ---
print("Initialisation du EndpointHandler...")
# Pour le test local, le token n'est pas nécessaire car le modèle est chargé depuis le Hub public.
# Si votre modèle était privé, vous devriez définir os.environ["HUGGING_FACE_HUB_TOKEN"] = "votre_token"
handler = EndpointHandler(path=MODEL_PATH_FOR_HANDLER)
print("EndpointHandler initialisé.")
# --- Préparation du payload (simule l'entrée de l'API) ---
print(f"Lecture de la vidéo de test: {VIDEO_FILE_PATH}")
try:
with open(VIDEO_FILE_PATH, "rb") as f:
video_bytes = f.read()
encoded_video = base64.b64encode(video_bytes).decode("utf-8")
except FileNotFoundError:
print(f"Erreur: Le fichier vidéo {VIDEO_FILE_PATH} n'a pas été trouvé.")
exit()
# Le payload doit correspondre à ce que votre handler.__call__ attend
# C'est-à-dire un dictionnaire avec une clé "inputs" qui contient une liste de messages
# et le contenu vidéo encodé en base64 sous la clé "data".
payload = {
"inputs": [
{
"role": "user",
"content": [
{"type": "text", "text": "Describe this video in detail"},
{"type": "video", "data": encoded_video} # Ici, on passe la vidéo encodée
]
}
],
"parameters": {
"max_new_tokens": 128 # Paramètres d'inférence optionnels
}
}
# --- Exécution de l'inférence ---
print("Lancement de l'inférence locale (cela peut prendre du temps)...")
try:
result = handler(payload)
print("\n--- Résultat de l'inférence ---")
print(result)
except Exception as e:
print(f"\nErreur lors de l'inférence: {e}") |