Update cluster_visualize.py

cluster_visualize.py

@@ -59,7 +59,7 @@ class PredictionArgs:
         self.head = head
         self.resize_img = resize_img
         self.alpha = alpha
-       # assert self.model in timm.list_models(), "Please use a timm pre-trined model, see timm.list_models()"
+        assert self.model in timm.list_models(), "Please use a timm pre-trined model, see timm.list_models()"
 
 # Preprocessing
 def _preprocess(raw_image):