brepositorium
commited on
Commit
·
ab79c8c
1
Parent(s):
6f0b3f6
fix model issues
Browse files- src/app.py +0 -1
src/app.py
CHANGED
|
@@ -7,7 +7,6 @@ import torch.nn as nn
|
|
| 7 |
from model import get_transforms, create_effnetb2_model
|
| 8 |
|
| 9 |
model = create_effnetb2_model(num_classes=3)
|
| 10 |
-
model.load_state_dict(torch.load('model.pth', map_location=torch.device('cpu')))
|
| 11 |
model.eval()
|
| 12 |
|
| 13 |
cns = ['negative', 'neutral', 'positive']
|
|
|
|
| 7 |
from model import get_transforms, create_effnetb2_model
|
| 8 |
|
| 9 |
model = create_effnetb2_model(num_classes=3)
|
|
|
|
| 10 |
model.eval()
|
| 11 |
|
| 12 |
cns = ['negative', 'neutral', 'positive']
|