fix: bugs

data_upload/data_upload_page.py

@@ -18,7 +18,8 @@ def data_upload(clip_model, preprocess, text_embedding_model):
                 with cols[count]:
                     st.image(image)
             with cols[4]:
-                st.info(f"and more {len(images) - 5} images...")
+                if len(images) > 5:
+                    st.info(f"and more {len(images) - 5} images...")
             st.info(f"Total {len(images)} files selected.")
             if st.button("Add Images"):
                 progress_bar = st.progress(0)

model_finetuning/components/model_training_component.py

@@ -34,6 +34,10 @@ def model_training():
     st.subheader("Data Preview")
     st.dataframe(annotations_df.head(), use_container_width=True)
 
+    if len(annotations_df) < 2:
+        st.error("Not enough data to train the model.")
+        return
+
     test_size = st.selectbox("Select Test Size", options=[0.1, 0.2, 0.3, 0.4, 0.5], index=1)
     train_df, val_df = train_test_split(annotations_df, test_size=test_size, random_state=42)
     if len(train_df) < 2: