Hugging Face's logo

Spaces:
DimaML
/
emotion-classifierV2
Sleeping

App Files Community
DimaML commited on
Commit
2fe11a9
·
verified ·
1 Parent(s): a7d5366

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -10,7 +10,7 @@ device = torch.device("cpu")
10
  class_names = ['Anger', 'Disgust', 'Fear', 'Happy', 'Pain', 'Sad']
11
  classes_count = len(class_names)
12
 
13
- model = model.renset18(weights='DEFAULT').to(device)
14
  model.fc = nn.Sequential(
15
  nn.Linear(512, classes_count)
16
  )
 
10
  class_names = ['Anger', 'Disgust', 'Fear', 'Happy', 'Pain', 'Sad']
11
  classes_count = len(class_names)
12
 
13
+ model = models.resnet18(weights='DEFAULT').to(device)
14
  model.fc = nn.Sequential(
15
  nn.Linear(512, classes_count)
16
  )