Update app.py

app.py

@@ -17,10 +17,8 @@ def preprocess(raw):
     image = image.reshape(image.shape[0]*image.shape[1], 3)
     return image
 
-def analyze(img):
-    modified_image = preprocess(img)
-    n_cluster = gr.inputs.Slider(minimum=2, maximum=10, step=1, label="Number of claster")
-    
+def analyze(img,n_cluster ):
+    modified_image = preprocess(img)    
     clf = KMeans(n_clusters = n_cluster)
     color_labels = clf.fit_predict(modified_image)
     center_colors = clf.cluster_centers_
@@ -38,5 +36,5 @@ def analyze(img):
     
     return plot
 
-color_picker = gr.Interface(fn=analyze, inputs="image", outputs="plot")
+color_picker = gr.Interface(fn=analyze, inputs=["image", gr.inputs.Slider(minimum=2, maximum=10, step=1, label="Number of claster")], outputs="plot")
 color_picker.launch()