Spaces:

zihaoz96
/

shark-classifier

Runtime error

zihaoz96 commited on Dec 10, 2021

Commit

626c0bb

1 Parent(s): def3421

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -20,7 +20,7 @@ def predict_image(image):
     image = np.expand_dims(image, axis=0)
     classifier = TorchVisionClassifierInference(
-        model_path = "./models/" + colname + ".pth",
     )
     pred = classifier.predict(img=image)

     image = np.expand_dims(image, axis=0)
     classifier = TorchVisionClassifierInference(
+        model_path = "./models/" + colname + "/best_model.pth",
     )
     pred = classifier.predict(img=image)