INNER CODE UNIT · Python

process_image_to_lottie

OpenVGLab/OmniLottie · app.py:645

def process_image_to_lottie(image_file, text_description, max_tokens, use_sampling, temperature, top_p, top_k):
    with generation_lock:
        try:
            start_time = time.time()

            if image_file is None:
                return None, "❌ Please upload an image", None

            model, processor, device = load_model_once()

            image = load_image_from_file(image_file)

            image = image.resize((448, 448), PILImage.LANCZOS)

            desc = text_description if text_description else "A simple animation"
            messages = build_messages("image", text_prompt=desc, image=image)
            inputs = prepare_inference_input(processor, messages, device)

View source record →