Hugging Face's logo

Spaces:
mhue
/
skin-scan
Sleeping

App Files Community
mhue commited on
Commit
504e97b
·
verified ·
1 Parent(s): 78548e0

Upload folder using huggingface_hub

Browse files
Files changed (3) hide show
  1. app.py +1 -1
  2. example1.jpg +0 -0
  3. example2.jpg +0 -0
app.py CHANGED
@@ -7,7 +7,7 @@ import timm
7
 
8
  # Load model
9
  model = timm.create_model("efficientnet_b3a", pretrained=True, num_classes=2)
10
- model.load_state_dict(torch.load("model.pth", map_location="cpu"))
11
  model.eval()
12
 
13
  # Transform
 
7
 
8
  # Load model
9
  model = timm.create_model("efficientnet_b3a", pretrained=True, num_classes=2)
10
+ # model.load_state_dict(torch.load("model.pth", map_location="cpu"))
11
  model.eval()
12
 
13
  # Transform
example1.jpg ADDED
example2.jpg ADDED