Update app.py

app.py

@@ -12,7 +12,7 @@ config = Config()
 config.HTMLExporter.preprocessors = ["nbconvert.preprocessors.ExtractOutputPreprocessor"]
 html_exporter = HTMLExporter(config=config, template_name="classic")
 
-ds = load_dataset("data-agents/kaggle-notebooks-edu-v0")
+ds = load_dataset("lvwerra/jupyter-notebooks-edu-v0")
 ds_out = ds.filter(lambda x: x["contains_outputs"])
 
 TMP_DIR = './tmp/'
@@ -46,7 +46,7 @@ def parse_notebook(score_options, output_options, index):
     else:
         sample = ds_out[score_options][index]
     
-    notebook_string = sample["text"]
+    notebook_string = sample["content"]
     notebook_response = sample["response"]
     notebook_id = sample["id"].split("/")[-1]