Commit
•
2733ea4
1
Parent(s):
bbfca17
Update model.py
Browse fileschange the model path
model.py
CHANGED
@@ -6,4 +6,4 @@ efficient_weight = EfficientNet_B2_Weights.DEFAULT
|
|
6 |
|
7 |
efficient_transformer = efficient_weight.transforms()
|
8 |
|
9 |
-
efficient_model = torch.load("
|
|
|
6 |
|
7 |
efficient_transformer = efficient_weight.transforms()
|
8 |
|
9 |
+
efficient_model = torch.load("efficient_b2.pth").to("cpu")
|