katsukiai commited on
Commit
7051fd2
·
verified ·
1 Parent(s): 73e2dba

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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="Katsukiai")]
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="Katsukiai")])
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"])