Update chatcode/build_notebook.py

chatcode/build_notebook.py

@@ -77,7 +77,7 @@ The model and datasets have their own pinned revisions in the configuration.
     from google.colab import userdata
     from huggingface_hub import HfApi, snapshot_download
     os.environ["HF_TOKEN"] = userdata.get("HF_TOKEN")
-    revision = SOURCE_REVISION or HfApi().model_info(SOURCE_REPO).sha
+    revision = HfApi().model_info(SOURCE_REPO, revision=SOURCE_REVISION or "main").sha
     source_dir = Path(snapshot_download(SOURCE_REPO, revision=revision, allow_patterns=[
         "app.py", "chat.py", "LICENSE*", "chatcode/*.py", "chatcode/*.json", "chatcode/*.txt", "chatcode/*.md"]))
     assert (source_dir / "chatcode/articles.py").is_file(), "This source commit predates article support."