Update app.py

app.py

@@ -14,7 +14,14 @@ def convert(token: str, model_id: str) -> str:
     try:
         api = HfApi(token=token)
         info = api.model_info(repo_id=model_id)
-        return info
+        
+        # loop info siblings (type is RepoFile) and find the ones which rfilename ends with .ckpt, then return them
+        ckpt_files = [sibling for sibling in info.siblings if sibling.rfilename.endswith(".ckpt")]
+        if len(ckpt_files) == 0:
+            return f"#### Error: No checkpoint file found in the model repo."
+        
+        # return file names:
+        return "\n".join([f"- {ckpt_file.rfilename}" for ckpt_file in ckpt_files])
         
     except Exception as e:
         return f"#### Error: {e}"