Update app.py

app.py

@@ -39,12 +39,12 @@ def update_max_index(score_option, output_option):
     return gr.Slider(maximum=max_index), gr.Number(value=0, maximum=max_index)
 
 
-def parse_notebook(filter_options, output_options, index):
+def parse_notebook(score_options, output_options, index):
     reset_tmp_folder()
     if output_options == "All":
-        sample = ds[i]
+        sample = ds[score_options][i]
     else:
-        sample = ds_out[i]
+        sample = ds_out[score_options][i]
     
     notebook_string = sample["text"]
     notebook_id = sample["id"].split("/")[-1]
@@ -83,7 +83,7 @@ with gr.Blocks() as demo:
         outputs=[index_slider, index_slider]
     )
     
-    index_slider.change(fn=parse_notebook, inputs=[filter_options,output_options, index_slider], outputs=[html, file])
-    demo.load(fn=parse_notebook, inputs=[filter_options,output_options, index_slider], outputs=[html, file])
+    index_slider.change(fn=parse_notebook, inputs=[score_options, output_options, index_slider], outputs=[html, file])
+    demo.load(fn=parse_notebook, inputs=[score_options, output_options, index_slider], outputs=[html, file])
 
 demo.launch()