New path

pages/Upload_and_Predict.py

@@ -7,6 +7,8 @@ import pandas as pd
 from collections import defaultdict
 import os
 from datasets import load_dataset
+import io
+import requests
 
 # Title
 st.markdown("<h2 style='color: #2E86C1;'>📷 Upload & Predict</h2>", unsafe_allow_html=True)
@@ -58,7 +60,7 @@ class_names = ['No DR', 'Mild', 'Moderate', 'Severe', 'Proliferative DR']
 
 @st.cache_data
 def load_sample_images_from_csv():
-    csv_url = "https://huggingface.co/datasets/Ci-Dave/DDR_dataset_train_test/tree/main/splits/test"
+    csv_url = "https://huggingface.co/datasets/Ci-Dave/DDR_dataset_train_test/raw/main/splits/test_labels.csv"
     df = pd.read_csv(csv_url)
     samples = defaultdict(list)
 

requirements.txt

@@ -10,3 +10,5 @@ torchvision
 scikit-learn
 fpdf
 datasets
+requests
+io