adikuma
/

Image-Colour-Extract

Model card Files Files and versions

ADITYA KUMAR commited on Aug 1, 2023

Commit

fb1e70e

·

1 Parent(s): 1bbd364

Update predict.py

Files changed (1) hide show

predict.py +2 -2

predict.py CHANGED Viewed

@@ -4,8 +4,8 @@ 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):

 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):