Spaces:
Runtime error
Runtime error
Added download_test_data method
Browse files- knowledge_gpt/core/data.py +12 -0
knowledge_gpt/core/data.py
ADDED
@@ -0,0 +1,12 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
import streamlit as st
|
2 |
+
from gdown import download_folder
|
3 |
+
|
4 |
+
|
5 |
+
@st.cache_resource(show_spinner=False)
|
6 |
+
def download_test_data():
|
7 |
+
# url = f"https://drive.google.com/drive/folders/uc?export=download&confirm=pbef&id={file_id}"
|
8 |
+
url = "https://drive.google.com/drive/folders/1uDSAWtLvp1YPzfXUsK_v6DeWta16pq6y"
|
9 |
+
with st.spinner(text="Downloading test data. This might take a minute."):
|
10 |
+
# @TODO: replace gown solution with a custom solution compatible with GitHub and
|
11 |
+
# use st.progress to get more verbose during download
|
12 |
+
download_folder(url=url, quiet=False, use_cookies=False, output="./data/")
|