Hugging Face's logo

Spaces:
Tanusree88
/
ViT-MRI-FineTuning
Sleeping

App Files Community
Tanusree88 commited on
Commit
68257a1
·
verified ·
1 Parent(s): c309443

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -86,7 +86,7 @@ def fine_tune_model(train_loader):
86
  st.title("Fine-tune ViT on MRI Scans")
87
 
88
  if st.button("Start Training"):
89
- extract_zip('your_zip_file.zip', 'extracted_folder/')
90
  image_paths, labels = prepare_dataset('extracted_folder/')
91
  dataset = CustomImageDataset(image_paths, labels)
92
  train_loader = DataLoader(dataset, batch_size=32, shuffle=True)
 
86
  st.title("Fine-tune ViT on MRI Scans")
87
 
88
  if st.button("Start Training"):
89
+ extract_zip('https://huggingface.co/spaces/Tanusree88/ViT-MRI-FineTuning/resolve/main/archive%20(5).zip', '')
90
  image_paths, labels = prepare_dataset('extracted_folder/')
91
  dataset = CustomImageDataset(image_paths, labels)
92
  train_loader = DataLoader(dataset, batch_size=32, shuffle=True)