test_handler.py

from handler import EndpointHandler

# Initialize handler
my_handler = EndpointHandler(path="DAMO-NLP-SG/VideoLLaMA2-8x7B")

# Prepare sample payloads
video_payload = {
    "paths": ["assets/cat_and_chicken.mp4"],
    "questions": ["What animals are in the video, what are they doing, and how does the video feel?"],
    "modal_list": ["video"]
}

image_payload = {
    "paths": ["assets/sora.png"],
    "questions": ["What is the woman wearing, what is she doing, and how does the image feel?"],
    "modal_list": ["image"]
}

# Test the handler
video_pred = my_handler(video_payload)
image_pred = my_handler(image_payload)

# Show results
print("video_pred", video_pred)
print("image_pred", image_pred)

# Expected Output Examples:
# video_pred [{'output': 'The video features a kitten and a baby chick playing together. The kitten is seen laying on the floor while the baby chick hops around. The two animals interact playfully with each other, and the video has a cute and heartwarming feel to it.'}]
# image_pred [{'output': 'The woman in the image is wearing a black coat and sunglasses, and she is walking down a rain-soaked city street. The image feels vibrant and lively, with the bright city lights reflecting off the wet pavement, creating a visually appealing atmosphere.'}]