Spaces:

bhushanp
/

car_classifier

Running

bhushanp commited on Mar 2

Commit

04f6df4

•

1 Parent(s): ac4b8ad

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -18,7 +18,7 @@ def get_label(fname):
 #     print(cls)
     return name(cls)
-learn = load_learner("./models/model.pth")
 # Function to make predictions from an image

 #     print(cls)
     return name(cls)
+learn = load_learner("./models/model.pkl")
 # Function to make predictions from an image