clementBE commited on
Commit
52e5e5d
·
verified ·
1 Parent(s): 2137e43

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -78,7 +78,7 @@ def closest_basic_color(rgb):
78
  return closest_color
79
 
80
  def get_dominant_color(image, num_colors=5):
81
- image = image.resize((300, 300))
82
  pixels = np.array(image).reshape(-1, 3)
83
  kmeans = MiniBatchKMeans(n_clusters=num_colors, random_state=0, n_init=5)
84
  kmeans.fit(pixels)
 
78
  return closest_color
79
 
80
  def get_dominant_color(image, num_colors=5):
81
+ image = image.resize((200, 200))
82
  pixels = np.array(image).reshape(-1, 3)
83
  kmeans = MiniBatchKMeans(n_clusters=num_colors, random_state=0, n_init=5)
84
  kmeans.fit(pixels)