Update app.py

app.py

@@ -5,8 +5,12 @@ from transformers import AutoTokenizer, AutoModelForSequenceClassification
 from sklearn.metrics.pairwise import cosine_similarity
 import torch
 import re
+from datasets import load_dataset
+
+dataset = load_dataset("JuliaWolken/chunks.csv", split="train")
+chunks_df = dataset.to_pandas()
+
 
-chunks_data_path = 'JuliaWolken/chunks-project'
 model = SentenceTransformer('JuliaWolken/fine_tuned_model_with_triplets')
 
 chunks_df = pd.read_csv(chunks_data_path)