Update app.py

app.py

@@ -46,9 +46,9 @@ def preprocess_image(image_path):
 def prepare_dataset(extracted_folder):
     image_paths = []
     labels = []
-    for disease_folder in ['alzheimers_dataset', 'parkinsons_dataset', 'ms']:
+    for disease_folder in ['alzheimers_dataset', 'parkinsons_dataset', 'MSjpg']:
         folder_path = os.path.join(extracted_folder, disease_folder)
-        label = {'alzheimers_dataset': 0, 'parkinsons_dataset': 1, 'ms': 2}[disease_folder]
+        label = {'alzheimers_dataset': 0, 'parkinsons_dataset': 1, 'MSjpg': 2}[disease_folder]
         for img_file in os.listdir(folder_path):
             if img_file.endswith(('.nii', '.jpg', '.jpeg')):
                 image_paths.append(os.path.join(folder_path, img_file))
@@ -141,9 +141,9 @@ def preprocess_image(image_path):
 def prepare_dataset(extracted_folder):
     image_paths = []
     labels = []
-    for disease_folder in ['alzheimers_dataset', 'parkinsons_dataset', 'ms']:
+    for disease_folder in ['alzheimers_dataset', 'parkinsons_dataset', 'MSjpg']:
         folder_path = os.path.join(extracted_folder, disease_folder)
-        label = {'alzheimers_dataset': 0, 'parkinsons_dataset': 1, 'ms': 2}[disease_folder]
+        label = {'alzheimers_dataset': 0, 'parkinsons_dataset': 1, 'MSjpg': 2}[disease_folder]
         for img_file in os.listdir(folder_path):
             if img_file.endswith(('.nii', '.jpg', '.jpeg')):
                 image_paths.append(os.path.join(folder_path, img_file))
@@ -188,7 +188,7 @@ def fine_tune_model(train_loader):
 # Streamlit UI for Fine-tuning
 st.title("Fine-tune ViT on MRI/CT Scans for MS & Neurodegenerative Diseases")
 
-zip_file_url = "https://huggingface.co/spaces/Tanusree88/ViT-MRI-FineTuning/resolve/main/archive%20(5).zip"
+zip_file_url = "https://huggingface.co/spaces/Tanusree88/ViT-MRI-FineTuning/resolve/main/neuroniiimages.zip"
 
 if st.button("Start Training"):
     extraction_dir = "extracted_files"