jbilcke-hf
/

LTX-Video-0.9.1-HFIE

Inference Endpoints

Model card Files Files and versions Community

jbilcke-hf HF staff commited on 14 days ago

Commit

3841f91

•

1 Parent(s): 201ac66

Update handler.py

Files changed (1) hide show

handler.py +1 -1

handler.py CHANGED Viewed

@@ -339,7 +339,7 @@ class EndpointHandler:
                 if input_image:
                     # Process base64 image
                     if input_image.startswith('data:'):
-                        input_image = image_data.split(',', 1)[1]
                     image_bytes = base64.b64decode(input_image)
                     image = Image.open(io.BytesIO(image_bytes)).convert("RGB")
                     generation_kwargs["image"] = image

                 if input_image:
                     # Process base64 image
                     if input_image.startswith('data:'):
+                        input_image = input_image.split(',', 1)[1]
                     image_bytes = base64.b64decode(input_image)
                     image = Image.open(io.BytesIO(image_bytes)).convert("RGB")
                     generation_kwargs["image"] = image