Spaces:

noelsinghsr
/

airplaneclassifier

Sleeping

noelsinghsr commited on Jun 20, 2023

Commit

5a754f4

•

1 Parent(s): d962a77

Update model.py

updated directory path of model file

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/airplanes/airplane_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('airplane_uncompiled.pth', map_location=torch.device('cpu'))
     model_weights = torchvision.models.EfficientNet_B2_Weights.DEFAULT
     transforms = model_weights.transforms()