Spaces:
Runtime error
Runtime error
Jason Adrian
commited on
Commit
•
e8376fa
1
Parent(s):
d14fd28
changes to app
Browse files
app.py
CHANGED
@@ -7,7 +7,7 @@ from resnet18 import ResNet18
|
|
7 |
|
8 |
model = ResNet18(1, 5)
|
9 |
|
10 |
-
checkpoint = torch.load('
|
11 |
|
12 |
# The state dict will contains net.layer_name
|
13 |
# Our model doesn't contains `net.` so we have to rename it
|
|
|
7 |
|
8 |
model = ResNet18(1, 5)
|
9 |
|
10 |
+
checkpoint = torch.load('acc=0.94.ckpt')
|
11 |
|
12 |
# The state dict will contains net.layer_name
|
13 |
# Our model doesn't contains `net.` so we have to rename it
|