Spaces:

IncreasingLoss
/

Wildlife_Animal_Classifier

Running

IncreasingLoss commited on Apr 14

Commit

0db077e

verified ·

1 Parent(s): dfef520

Upload app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -133,7 +133,8 @@ class MobileNetV3_Small(nn.Module):
 """creating modelinstance"""
 model = MobileNetV3_Small().to("cpu")
-model.load_state_dict( torch.load("MobileNet3_small_full.pth"))
 classes = ['antelope',
  'buffalo',
  'chimpanzee',

 """creating modelinstance"""
 model = MobileNetV3_Small().to("cpu")
+model.load_state_dict( torch.load("MobileNet3_small_StateDictionary.pth"))
 classes = ['antelope',
  'buffalo',
  'chimpanzee',