Spaces:

re-mind
/

Similarity_Search

Running

amaye15 commited on Jan 13

Commit

0ba3d8c

1 Parent(s): e0b1978

Debug - Remove Column Removal

Files changed (1) hide show

src/api/services/huggingface_service.py CHANGED Viewed

@@ -16,7 +16,7 @@ class HuggingFaceService:
         """Push the dataset to Hugging Face Hub."""
         try:
             logger.info(f"Creating Hugging Face Dataset: {dataset_name}...")
-            ds = Dataset.from_pandas(df).remove_columns("__index_level_0__")
             ds.push_to_hub(dataset_name)
             logger.info(f"Dataset pushed to Hugging Face Hub: {dataset_name}")
         except Exception as e:

         """Push the dataset to Hugging Face Hub."""
         try:
             logger.info(f"Creating Hugging Face Dataset: {dataset_name}...")
+            ds = Dataset.from_pandas(df)  # .remove_columns("__index_level_0__")
             ds.push_to_hub(dataset_name)
             logger.info(f"Dataset pushed to Hugging Face Hub: {dataset_name}")
         except Exception as e: