Spaces:

leynessa
/

libkamaja_id

Sleeping

leynessa commited on Jul 5

Commit

61b15eb

verified ·

1 Parent(s): d164506

Update streamlit_app.py

Files changed (1) hide show

streamlit_app.py CHANGED Viewed

@@ -20,7 +20,7 @@ except:
 def load_model():
     model = models.resnet18(pretrained=False)
     model.fc = torch.nn.Linear(model.fc.in_features, len(class_names))
-    model.load_state_dict(torch.load("/content/drive/MyDrive/butterfly_classifier.pth", map_location="cpu"))
     model.eval()
     return model

 def load_model():
     model = models.resnet18(pretrained=False)
     model.fc = torch.nn.Linear(model.fc.in_features, len(class_names))
+    model.load_state_dict(torch.load("butterfly_classifier.pth", map_location="cpu"))
     model.eval()
     return model