rahideer commited on
Commit
3e7f650
·
verified ·
1 Parent(s): 6273efa

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -5,8 +5,8 @@ from transformers import RagTokenizer, RagRetriever, RagSequenceForGeneration
5
  # Load a multilingual dataset (xnli or tydi_qa)
6
  def load_data():
7
  try:
8
- # Load the 'xnli' dataset, validation split
9
- dataset = load_dataset("xnli", split="validation")
10
  st.write(f"Loaded {len(dataset)} examples from the 'validation' split.")
11
  return dataset
12
  except Exception as e:
 
5
  # Load a multilingual dataset (xnli or tydi_qa)
6
  def load_data():
7
  try:
8
+ # Use a specific version of the dataset
9
+ dataset = load_dataset("xnli", "all_languages", split="validation") # Using a direct name instead of a wildcard pattern
10
  st.write(f"Loaded {len(dataset)} examples from the 'validation' split.")
11
  return dataset
12
  except Exception as e: