Update app.py

app.py

@@ -69,11 +69,11 @@ def load_movie_data():
     try:
         dataset = load_dataset(
             "facebook/wiki_movies", 
-            split="train[:2000]", 
+            split="train", 
             trust_remote_code=True  # Explicitly allow trusted code
         )
         df = pd.DataFrame(dataset)
-         df['context'] = "Question: " + df['question'].str.strip() + "\n" + \
+        df['context'] = "Question: " + df['question'].str.strip() + "\n" + \
                         "Answer: " + df['answer'].str.strip()
         return df