| from transformers import pipeline, AutoTokenizer, AutoModelForSequenceClassification |
| from PIL import Image |
| from io import BytesIO |
| import requests |
| import numpy as np |
|
|
| model = AutoModelForSequenceClassification.from_pretrained("./extract-colors.py") |
| tokenizer = AutoTokenizer.from_pretrained("./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 |
|
|
