| from transformers import pipeline, AutoTokenizer, AutoModelForSequenceClassification | |
| from PIL import Image | |
| from io import BytesIO | |
| import requests | |
| import numpy as np | |
| model = AutoModelForSequenceClassification.from_pretrained("./model/extract-colors.py") | |
| tokenizer = AutoTokenizer.from_pretrained("./model/tokenizer", tokenizer_func=tokenizer_function) | |
| model_class = pipeline('image-classification', model=model, tokenizer=tokenizer) | |
| def get_colors_and_closest_to_white(image_url): | |
| response = requests.get(image_url) | |
| img = Image.open(BytesIO(response.content)) | |
| img_array = np.array(img) | |
| result = model_class(image_array=img_array) | |
| return result | |