Update app/main.py

app/main.py

@@ -116,7 +116,7 @@ def chunk_embed(input_ds, input_splits, input_text_col, chunk_out_ds,
     gr.Info("Started chunking")
     try:
         input_splits = [spl.strip() for spl in input_splits.split(",") if spl]
-        input_ds = load_dataset(input_ds, split="+".join(input_splits), token=HF_TOKEN)
+        input_ds = load_dataset(input_ds, "text-corpus", split="+".join(input_splits), token=HF_TOKEN)
         chunker = Chunker(strategy, split_seq, chunk_len)
     except Exception as e:
         raise gr.Error(str(e))