Hi! I recently hosted " [runwayml/stable-diffusion-v1-5]" on the InferenceApi. I am trying to hit the endpoint and store the resulting image locally. My code is as follows:
def client(): token = '<my_token>' client = InferenceClient( model="<my_model_endpoint>", token=token ) image = client.text_to_image("An astronaut riding a horse on the moon.") image.save("astronaut.png")
However I keep getting this error output:
raise UnidentifiedImageError(msg) PIL.UnidentifiedImageError: cannot identify image file <_io.BytesIO object at 0x1042384f0>
Any help here would be much appreciated!