it needs to be able to see a function named main

app/download_models.py

@@ -1,11 +1,12 @@
 import os
 from huggingface_hub import hf_hub_download
 
-# Get the Hugging Face token from environment variable (automatically provided from the secret)
-HUGGINGFACE_TOKEN = os.getenv("HF_TOKEN")
 
-if HUGGINGFACE_TOKEN is None:
-    raise ValueError("Hugging Face token is missing. Please set the 'HF_TOKEN' environment variable.")
+# download_models.py
+
+import os
+from huggingface_hub import hf_hub_download
+
 
 def create_directory(dir_path):
     """
@@ -40,7 +41,15 @@ def download_model(repo_id, filename, destination_dir, token):
     return full_path
 
 
-if __name__ == "__main__":
+def main():
+    # Get the Hugging Face token from environment variable (automatically provided from the secret)
+    HUGGINGFACE_TOKEN = os.getenv("HF_TOKEN")
+
+    if HUGGINGFACE_TOKEN is None:
+        raise ValueError(
+            "Hugging Face token is missing. Please set the 'HF_TOKEN' environment variable."
+        )
+
     # Define the repositories and filenames
     magic_gnn_repo = "uripper/magic-gnn"
     oracle_cards_graph_repo = "uripper/oracle_cards_graph"
@@ -70,3 +79,7 @@ if __name__ == "__main__":
     print("\nDownload complete:")
     print(f"Card Graph: {card_graph_path}")
     print(f"Oracle Cards Graph: {oracle_cards_graph_path}")
+
+
+if __name__ == "__main__":
+    main()