Update app.py

app.py

@@ -1,15 +1,51 @@
+from huggingface_hub import hf_hub_download
 import json
-import gradio as gr
+import pandas as pd
 from pathlib import Path
 
-DATA_FILE = "EIRKG_v1.1_full.json"
+REPO_ID = "ashutoshzade/EIRKG"
 
-# ---------------------
-# Load dataset
-# ---------------------
+# CHANGE THIS TO YOUR ACTUAL FILE
+FILENAME = "YOUR_JSON_FILENAME.json"
+
+
+def load_dataset():
+
+    try:
+
+        print("Downloading EIRKG dataset...")
+
+        path = hf_hub_download(
+            repo_id=REPO_ID,
+            repo_type="dataset",
+            filename=FILENAME
+        )
+
+        print("Success!")
+
+        if FILENAME.endswith(".json"):
+            with open(path,"r",encoding="utf-8") as f:
+                return json.load(f)
+
+        if FILENAME.endswith(".jsonl"):
+            data=[]
+            with open(path,"r",encoding="utf-8") as f:
+                for line in f:
+                    data.append(json.loads(line))
+            return data
+
+        if FILENAME.endswith(".csv"):
+            return pd.read_csv(path).to_dict("records")
+
+    except Exception as e:
+
+        print(e)
+
+        return []
+
+DATA = load_dataset()
 
-with open(DATA_FILE, "r", encoding="utf-8") as f:
-    DATA = json.load(f)
+print(len(DATA))
 
 # ---------------------
 # Build searchable text