gnikhilchand commited on
Commit
d73f407
·
verified ·
1 Parent(s): 1a75b1d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -17,7 +17,7 @@ model.fc = nn.Linear(num_ftrs, 2)
17
 
18
  # 3. LOAD WEIGHTS
19
  # Replace 'fire_detection_resnet18.pth' with the EXACT name of the file you uploaded
20
- model_path = "fire_detection_resnet18.pth"
21
 
22
  # Load weights for CPU (since HF Spaces Free Tier uses CPU)
23
  state_dict = torch.load(model_path, map_location=torch.device('cpu'))
 
17
 
18
  # 3. LOAD WEIGHTS
19
  # Replace 'fire_detection_resnet18.pth' with the EXACT name of the file you uploaded
20
+ model_path = "fire_detection_resnet50.pth"
21
 
22
  # Load weights for CPU (since HF Spaces Free Tier uses CPU)
23
  state_dict = torch.load(model_path, map_location=torch.device('cpu'))