TimStats commited on
Commit
dc0111e
·
verified ·
1 Parent(s): 8a361d9

Update app.R

Browse files
Files changed (1) hide show
  1. app.R +7 -25
app.R CHANGED
@@ -67,13 +67,15 @@ download_private_csv <- function(repo_id, filename) {
67
  }
68
 
69
  download_private_parquet <- function(repo_id, filename) {
 
 
 
 
 
70
 
71
  # Create a temporary file
72
  temp_file <- tempfile(fileext = ".parquet")
73
 
74
- # Alternative LFS URL format (more direct download)
75
- url <- paste0("https://huggingface.co/datasets/", repo_id, "/lfs/resolve/main/", filename, "?download=true")
76
-
77
  # Download directly to file
78
  response <- GET(
79
  url,
@@ -93,28 +95,8 @@ download_private_parquet <- function(repo_id, filename) {
93
  stop(paste("Error reading parquet file:", e$message))
94
  })
95
  } else {
96
- # Try the standard URL if LFS URL failed
97
- url <- paste0("https://huggingface.co/datasets/", repo_id, "/resolve/main/", filename)
98
-
99
- response <- GET(
100
- url,
101
- add_headers(Authorization = paste("Bearer", Sys.getenv("GETCSV"))),
102
- write_disk(temp_file, overwrite = TRUE)
103
- )
104
-
105
- if (status_code(response) == 200) {
106
- tryCatch({
107
- data <- read_parquet(temp_file)
108
- file.remove(temp_file)
109
- return(data)
110
- }, error = function(e) {
111
- file.remove(temp_file)
112
- stop(paste("Error reading parquet file:", e$message))
113
- })
114
- } else {
115
- file.remove(temp_file)
116
- stop(paste("Failed to download file. Status code:", status_code(response)))
117
- }
118
  }
119
  }
120
 
 
67
  }
68
 
69
  download_private_parquet <- function(repo_id, filename) {
70
+ library(httr)
71
+ library(arrow)
72
+
73
+ # Create the direct download URL based on your example
74
+ url <- paste0("https://huggingface.co/datasets/", repo_id, "/resolve/main/", filename, "?download=true")
75
 
76
  # Create a temporary file
77
  temp_file <- tempfile(fileext = ".parquet")
78
 
 
 
 
79
  # Download directly to file
80
  response <- GET(
81
  url,
 
95
  stop(paste("Error reading parquet file:", e$message))
96
  })
97
  } else {
98
+ file.remove(temp_file)
99
+ stop(paste("Failed to download file. Status code:", status_code(response)))
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
100
  }
101
  }
102