Spaces:

Fareskh12
/

Classifiers

Sleeping

Fareskh12 commited on Mar 24, 2025

Commit

6edd3ec

verified ·

1 Parent(s): 9b2517d

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -12,6 +12,7 @@ def single_image(image,mode):
         return np.array(img.convert('HSV')).flatten()
     else:
         kmeans = sio.load('Models/kmeans.skops')
         sift = cv2.SIFT_create()
         img = np.array(img.convert("L"))
         keypoints, descriptors = sift.detectAndCompute(img, None)

         return np.array(img.convert('HSV')).flatten()
     else:
         kmeans = sio.load('Models/kmeans.skops')
+        num_clusters=kmeans.n_clusters
         sift = cv2.SIFT_create()
         img = np.array(img.convert("L"))
         keypoints, descriptors = sift.detectAndCompute(img, None)