Function read_image

demo/script.py:186–193

(image_file)

    return model, tokenizer

def read_image(image_file):
    if image_file.startswith('http') or image_file.startswith('https'):
        response = requests.get(image_file)
        image = Image.open(BytesIO(response.content)).convert('RGB')
    else:
        image = Image.open(image_file).convert('RGB')
    
    return image

def load_image(image, input_size=224):
    crop_pct = 224 / 256