File size: 1,584 Bytes
8773ff3 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 |
import streamlit as st
from defaults import PROJECT_NAME, ARGILLA_URL, DATASET_REPO_ID
from utils import project_sidebar
from hub import push_argilla_dataset_to_hub
st.set_page_config(
page_title="Domain Data Grower",
page_icon="🧑🌾",
)
project_sidebar()
################################################################################
# HEADER
################################################################################
st.header("🧑🌾 Domain Data Grower")
st.divider()
st.write(
"""Once you have reviewed the synthetic data in Argilla, you can publish the
generated dataset to the Hub."""
)
################################################################################
# Configuration
################################################################################
st.divider()
st.write("🔬 Argilla API details to push the generated dataset")
argilla_url = st.text_input("Argilla API URL", ARGILLA_URL)
argilla_api_key = st.text_input("Argilla API Key", "owner.apikey")
argilla_dataset_name = st.text_input("Argilla Dataset Name", PROJECT_NAME)
dataset_repo_id = st.text_input("Dataset Repo ID", DATASET_REPO_ID)
st.divider()
if st.button("🚀 Publish the generated dataset"):
with st.spinner("Publishing the generated dataset..."):
push_argilla_dataset_to_hub(
name=argilla_dataset_name,
repo_id=dataset_repo_id,
url=argilla_url,
api_key=argilla_api_key,
workspace="admin",
)
st.success("The generated dataset has been published to the Hub.")
|