Hugging Face's logo

Spaces:
Shchushch
/
CV
Sleeping

App Files Community
Shchushch commited on
Commit
2b3401f
•
1 Parent(s): 7ad97f7

Update assets/skin.py

Browse files
Files changed (1) hide show
  1. assets/skin.py +1 -1
assets/skin.py CHANGED
@@ -3,7 +3,7 @@ import torch
3
  from torchvision import transforms
4
  from torchvision.models import VGG19_BN_Weights, vgg19_bn
5
  DEVICE = 'cpu'
6
- model =torch.load('assets/skin.pth', map_location=)
7
  skin_map={0:'Добро',1:'Зло'}
8
  def get_evil(img):
9
 
 
3
  from torchvision import transforms
4
  from torchvision.models import VGG19_BN_Weights, vgg19_bn
5
  DEVICE = 'cpu'
6
+ model =torch.load('assets/skin.pth', map_location=DEVICE)
7
  skin_map={0:'Добро',1:'Зло'}
8
  def get_evil(img):
9