update explore and apply

apply_model.py

@@ -118,6 +118,12 @@ if st.session_state.embeddings_df is not None and svm_clf is not None:
         apply_model = st.form_submit_button("Apply Model")
 
     if apply_model:
+        if 'Test' in st.session_state.embeddings_df.index:
+            test_videos = True
+        elif 'Images' in st.session_state.embeddings_df.index:
+            test_videos = True
+        else:
+            test_videos = False
         kwargs = {'embeddings_df' : st.session_state.embeddings_df, 
                 'specified_classes' : specified_classes,
                 'classes_to_remove' : None,
@@ -125,7 +131,7 @@ if st.session_state.embeddings_df is not None and svm_clf is not None:
                 'animal_state' : None,
                 'view' : None,
                 'shuffle_data' : False,
-                'test_videos' : list(set(st.session_state.embeddings_df['Source'].to_list()))}
+                'test_videos' : test_videos}
         train_embeds, train_labels, train_images, test_embeds, test_labels, test_images =\
             process_dataset_in_mem(**kwargs)
         

explore.py

@@ -177,7 +177,7 @@ st.subheader("generate or import embeddings")
 
 st.text("Upload files to generate embeddings.")
 with st.form('embedding_generation_settings'):
-    seq_file = st.file_uploader("Choose a video file", type=['seq', 'mp4'], accept_multiple_files=True)
+    seq_file = st.file_uploader("Choose a video file", type=['seq', 'mp4'], accept_multiple_files=False)
     annot_files = st.file_uploader("Choose an annotation File", type=['annot','csv'], accept_multiple_files=True)
     downsample_rate = st.number_input('Downsample Rate',value=4)
     submit_embed_settings = st.form_submit_button('Create Embeddings', type='secondary')
@@ -211,7 +211,7 @@ else:
 st.divider()
 st.subheader("provide video file if not yet already provided")
 
-uploaded_file = st.file_uploader("Choose a video file", type=['seq', 'mp4'], accept_multiple_files=True)
+uploaded_file = st.file_uploader("Choose a video file", type=['seq', 'mp4'], accept_multiple_files=False)
 
 st.divider()
 if st.session_state.embeddings_df is not None and (uploaded_file is not None or seq_file is not None):