File size: 666 Bytes
1bbd364 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 |
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
|