Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -123,7 +123,7 @@ class ImprovedUNet3D(nn.Module):
|
|
| 123 |
|
| 124 |
# --- Load Trained Model ---
|
| 125 |
# model_path = "/content/drive/MyDrive/BrainTumorBrats/final_trained_models/torch_save/brain_tumor_seg_final_dice_0.8814_epoch_99_complete_20250319_125101.pt"
|
| 126 |
-
model_path = "
|
| 127 |
device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
|
| 128 |
|
| 129 |
|
|
|
|
| 123 |
|
| 124 |
# --- Load Trained Model ---
|
| 125 |
# model_path = "/content/drive/MyDrive/BrainTumorBrats/final_trained_models/torch_save/brain_tumor_seg_final_dice_0.8814_epoch_99_complete_20250319_125101.pt"
|
| 126 |
+
model_path = "brain_tumor_seg_final_dice_0.8814_epoch_99_complete_20250319_125101.pt" # Change path if needed
|
| 127 |
device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
|
| 128 |
|
| 129 |
|