Update app.py

app.py

@@ -22,6 +22,8 @@ def main():
         folder_path = 'WayveScenes/'
     else:
         folder_path = ''
+    # Load the selected dataset
+    df = load_hf_datasets(dataset_name)
     # Input search query
     query = st.text_input("Enter your search query")
 
@@ -34,8 +36,6 @@ def main():
         if not query:
             st.warning("Please enter a search query.")
         else:
-            # Load the selected dataset
-            df = load_hf_datasets(dataset_name)
 
             # Perform the search
             results = search(query, df, limit, 0, "cosine", search_in_images=True, search_in_small_objects=False)