Spaces:

noelsinghsr
/

testspace

Sleeping

noelsinghsr commited on Jun 12, 2023

Commit

765f033

1 Parent(s): 3c94f3b

Update model.py

Files changed (1) hide show

model.py CHANGED Viewed

@@ -4,7 +4,7 @@ from torch import nn
 def load_model():
-    loaded_model = torch.load('demo/cifar10/cifar10_effnet_sm_uncompiled.pth', map_location=torch.device('cpu'))
     model_weights = torchvision.models.EfficientNet_B2_Weights.DEFAULT
     transforms = model_weights.transforms()

 def load_model():
+    loaded_model = torch.load('cifar10/cifar10_effnet_sm_uncompiled.pth', map_location=torch.device('cpu'))
     model_weights = torchvision.models.EfficientNet_B2_Weights.DEFAULT
     transforms = model_weights.transforms()