Spaces:

iisadia
/

Movie_Buff_QA

Sleeping

iisadia commited on Apr 13

Commit

bd86d68

verified ·

1 Parent(s): 472db3e

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -66,7 +66,7 @@ st.markdown("""
 def load_movie_data():
     try:
         # Try loading wiki_movies dataset
-        dataset = load_dataset("movie_rationales")
         df = pd.DataFrame(dataset)
         # Create synthetic movie data from Wikipedia snippets

 def load_movie_data():
     try:
         # Try loading wiki_movies dataset
+        dataset = load_dataset("wiki_movies", split="train[:1000]", trust_remote_code=True)
         df = pd.DataFrame(dataset)
         # Create synthetic movie data from Wikipedia snippets