Spaces:

cls00
/

tutorial

Runtime error

cls00 commited on Feb 23, 2024

Commit

d4e945d

verified ·

1 Parent(s): ba5d194

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -14,7 +14,7 @@ pathlib.WindowsPath = pathlib.PosixPath
 learn = load_learner('resnet50_10epochs.pkl')
 # %% ../app.ipynb 6
-categories = ['city','town','countryside','ocean']
 def classify_image(img):
     pred, idx, probs = learn.predict(img)

 learn = load_learner('resnet50_10epochs.pkl')
 # %% ../app.ipynb 6
+categories = ['city','countryside','ocean','town']
 def classify_image(img):
     pred, idx, probs = learn.predict(img)