Update app.py

app.py

@@ -5,6 +5,7 @@ import faiss
 import pickle
 from sentence_transformers import SentenceTransformer
 from groq import Groq
+from datasets import load_dataset
 
 # Load environment variables
 from dotenv import load_dotenv
@@ -17,8 +18,8 @@ MODEL_NAME = "llama-3-70b-8192"  # Or use "llama-3-8b-8192", "llama-3-3b-8192"
 # Load dataset
 @st.cache_data
 def load_data():
-    url = "https://huggingface.co/datasets/llmware/rag_instruct_benchmark_tester/resolve/main/rag_instruct_benchmark_tester.csv"
-    df = pd.read_csv(url)
+    dataset = load_dataset("llmware/rag_instruct_benchmark_tester", split="train")
+    df = pd.DataFrame(dataset)
     return df
 
 # Build or load FAISS index