dorsar commited on
Commit
15814eb
·
1 Parent(s): e3e80d7

update model path

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -7,7 +7,7 @@ import gradio as gr
7
  device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
8
 
9
  model = ResNetLungCancer(num_classes=4)
10
- model.load_state_dict(torch.load('Model/lung_cancer_detection_model.pth', map_location=device))
11
  model = model.to(device)
12
  model.eval()
13
 
 
7
  device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
8
 
9
  model = ResNetLungCancer(num_classes=4)
10
+ model.load_state_dict(torch.load('lung_cancer_detection_model.pth', map_location=device))
11
  model = model.to(device)
12
  model.eval()
13