Spaces:

noelsinghsr
/

testspace

Sleeping

noelsinghsr commited on Jun 12, 2023

Commit

644d850

1 Parent(s): 765f033

Update model.py

updated model path

Files changed (1) hide show

model.py CHANGED Viewed

@@ -4,7 +4,7 @@ from torch import nn
 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()

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