SUNGJIN LEE commited on
Commit
588895d
1 Parent(s): 2ac1269

Huggingface 토큰 추가

Browse files
Files changed (1) hide show
  1. data.py +4 -1
data.py CHANGED
@@ -1,6 +1,9 @@
1
  from datasets import load_dataset
2
  import streamlit as st
3
  import pandas as pd
 
 
 
4
 
5
  data_files = {
6
  "train": "ELG_Busan_PoC_per_CA_site_0226_0407.csv",
@@ -10,7 +13,7 @@ data_files = {
10
  @st.cache_data(show_spinner=False)
11
  def load_data():
12
 
13
- dataset = load_dataset('skt-asap/busan-poc-dataset', data_files=data_files)
14
  df = dataset['train'].to_pandas()
15
 
16
  df['ru_svc_lat_val'] = df['ru_svc_lat_val'].astype(float)
 
1
  from datasets import load_dataset
2
  import streamlit as st
3
  import pandas as pd
4
+ import os
5
+
6
+ token = os.environ['HF_TOKEN']
7
 
8
  data_files = {
9
  "train": "ELG_Busan_PoC_per_CA_site_0226_0407.csv",
 
13
  @st.cache_data(show_spinner=False)
14
  def load_data():
15
 
16
+ dataset = load_dataset('skt-asap/busan-poc-dataset', data_files=data_files, token=token)
17
  df = dataset['train'].to_pandas()
18
 
19
  df['ru_svc_lat_val'] = df['ru_svc_lat_val'].astype(float)