Update app.py

app.py

@@ -2,6 +2,8 @@ import os
 import json
 import requests
 
+import numpy as np
+
 import gradio as gr
 import pandas as pd
 from huggingface_hub import HfApi, hf_hub_download, snapshot_download
@@ -266,10 +268,12 @@ def filter_data(rl_env, task_id, selected_values, path):
     Filters the data based on the selected difficulty/pattern values.
     """
     data = get_data(rl_env, task_id, path)
-    
+
     # If there are selected values, filter the DataFrame
     if selected_values:
         filter_column = "Pattern" if "Pattern" in data.columns else "Difficulty"
+        if filter_column == "Difficulty":
+            selected_values = [np.int64(sv) for sv in selected_values] 
         data = data[data[filter_column].isin(selected_values)]
 
     return data