Update app.py

app.py

@@ -68,13 +68,13 @@ def load_movie_data():
     # Option 1: Try loading with trust_remote_code
     try:
         dataset = load_dataset(
-            "eraser-benchmark/movie_rationales", 
+            "facebook/wiki_movies", 
             split="train[:2000]", 
             trust_remote_code=True  # Explicitly allow trusted code
         )
         df = pd.DataFrame(dataset)
-        df['context'] = "Review: " + df['review'].str.strip() + "\n" + \
-                        "Label: " + df['label'].astype(str)
+         df['context'] = "Question: " + df['question'].str.strip() + "\n" + \
+                        "Answer: " + df['answer'].str.strip()
         return df
         
     except Exception as e: