Update app.py

app.py

@@ -57,7 +57,7 @@ for repo_id in repo_ids:
         )[-1]
         ckpt_name = f"G_{latest_id}.pth"
 
-    cluster_model_name = "kmeans.pt"
+    cluster_model_name = cluster_model_name or "kmeans.pt"
     if cluster_model_name in list_repo_files(repo_id):
         print(f"Found Cluster model - Downloading {cluster_model_name} from {repo_id}")
         cluster_model_path = hf_hub_download(repo_id, cluster_model_name)