Update download_data.R

download_data.R

@@ -1,18 +1,20 @@
 library(httr)
 
 token    <- Sys.getenv("HF_TOKEN")
-repo     <- "BU-Baseball/butler-baseball-data"
+repo     <- "BU-Baseball/BU-Baseball-Data"
 base_url <- paste0("https://huggingface.co/datasets/", repo, "/resolve/main/")
 
-download_hf <- function(filename) {
-  dest <- paste0("/app/", filename)
+download_hf <- function(filename, url_filename = NULL) {
+  dest         <- paste0("/app/", filename)
+  url_filename <- if (is.null(url_filename)) filename else url_filename
+
   if (file.exists(dest)) {
     message(filename, " already exists, skipping.")
     return(invisible(NULL))
   }
   message("Downloading ", filename, "...")
   resp <- GET(
-    paste0(base_url, filename),
+    paste0(base_url, url_filename),
     add_headers(Authorization = paste("Bearer", token)),
     write_disk(dest, overwrite = TRUE),
     progress()
@@ -22,12 +24,10 @@ download_hf <- function(filename) {
   message(filename, " done.")
 }
 
-files <- c(
-  "Stuff+2.rds",
-  "NEW_LeagueStats2.rds",
-  "location_plus_model.rds",
-  "location_plus_league_stats_pitcher.rds"
-)
+# Stuff+2.rds is encoded as Stuff%2B2.rds in the URL
+download_hf("Stuff+2.rds",                        "Stuff%2B2.rds")
+download_hf("NEW_LeagueStats2.rds")
+download_hf("location_plus_model.rds")
+download_hf("location_plus_league_stats_pitcher.rds")
 
-for (f in files) download_hf(f)
 message("All model files ready.")