Gajendra5490 commited on
Commit
70f1082
·
verified ·
1 Parent(s): 45003ca

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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 = "final_trained_models/torch_save/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
 
 
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