Update app.py

app.py

@@ -2,14 +2,17 @@ import torch
 import gradio as gr
 from fastai.text.all import *
 from transformers import BartTokenizer
-from datasets import load_metric
+from datasets import load_dataset, load_metric
 
 # Load the pre-trained model and tokenizer (adjust for Bart if needed)
 pretrained_model_name = "facebook/bart-large-cnn"  # Or "facebook/bart-base"
 hf_tokenizer = BartTokenizer.from_pretrained(pretrained_model_name)
 
+# Load the SQuAD dataset
+squad = load_dataset("squad")
+
 # Load the SQuAD metric
-squad_metric = load_metric("squad")
+squad_metric = load_metric("squad", data_dir="path/to/squad/dataset", trust_remote_code=True)
 
 def summarize(article):
     # Define your data transformation pipeline here, if applicable