Update app.py
Browse files
app.py
CHANGED
@@ -87,7 +87,7 @@ def train_with_deepseek(dataset, model_name="deepseek-ai/DeepSeek-V3"):
|
|
87 |
if selected_tab == "Train":
|
88 |
st.title("Train Katsukiai Dataset")
|
89 |
api = HfApi()
|
90 |
-
datasets_list = [d.id.split("/")[-1] for d in api.list_datasets(author="
|
91 |
dataset_name = st.selectbox("Select Dataset", datasets_list)
|
92 |
if st.button("Start Training"):
|
93 |
dataset = load_katsukiai_dataset(dataset_name)
|
@@ -101,7 +101,7 @@ if selected_tab == "Train":
|
|
101 |
|
102 |
elif selected_tab == "Train with DeepSeek-V3":
|
103 |
st.title("Train with DeepSeek-V3")
|
104 |
-
dataset_name = st.selectbox("Select Dataset", [d.id.split("/")[-1] for d in api.list_datasets(author="
|
105 |
if st.button("Train with DeepSeek"):
|
106 |
if st.session_state.settings["token"]:
|
107 |
login(st.session_state.settings["token"])
|
|
|
87 |
if selected_tab == "Train":
|
88 |
st.title("Train Katsukiai Dataset")
|
89 |
api = HfApi()
|
90 |
+
datasets_list = [d.id.split("/")[-1] for d in api.list_datasets(author="katsukiai")]
|
91 |
dataset_name = st.selectbox("Select Dataset", datasets_list)
|
92 |
if st.button("Start Training"):
|
93 |
dataset = load_katsukiai_dataset(dataset_name)
|
|
|
101 |
|
102 |
elif selected_tab == "Train with DeepSeek-V3":
|
103 |
st.title("Train with DeepSeek-V3")
|
104 |
+
dataset_name = st.selectbox("Select Dataset", [d.id.split("/")[-1] for d in api.list_datasets(author="katsukiai")])
|
105 |
if st.button("Train with DeepSeek"):
|
106 |
if st.session_state.settings["token"]:
|
107 |
login(st.session_state.settings["token"])
|