Spaces:

Conn-Finnegan
/

Soma_Skin_Cancer_Classifier

Sleeping

Conn Finnegan commited on May 1, 2025

Commit

b32662e

verified ·

1 Parent(s): 4d4e023

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -6,7 +6,7 @@ from PIL import Image
 # Load model
 model = models.resnet18()
 model.fc = torch.nn.Linear(model.fc.in_features, 2)
-model.load_state_dict(torch.load("skin_cancer_resnet18_v1.pt", map_location="cpu"))
 model.eval()
 # Classes

 # Load model
 model = models.resnet18()
 model.fc = torch.nn.Linear(model.fc.in_features, 2)
+model.load_state_dict(torch.load("skin_cancer_resnet18_version1.pt", map_location="cpu"))
 model.eval()
 # Classes