Update app.py

app.py

@@ -72,6 +72,7 @@ def load_movie_data():
             split="train[:2000]", 
             trust_remote_code=True  # Explicitly allow trusted code
         )
+        print(dataset.column_names)
         df = pd.DataFrame(dataset)
         df['context'] = "Title: " + df['movie'] + "\n" + \
                        "Plot: " + df['plot'] + "\n" + \