Update app.py
Browse files
app.py
CHANGED
|
@@ -7,7 +7,7 @@ neural_net = load_learner('trained-NN.pkl')
|
|
| 7 |
labels = neural_net.dls.vocab
|
| 8 |
|
| 9 |
def predict(img):
|
| 10 |
-
img =
|
| 11 |
category, idx, probs = neural_net.predict(img)
|
| 12 |
return 'code has run'
|
| 13 |
|
|
|
|
| 7 |
labels = neural_net.dls.vocab
|
| 8 |
|
| 9 |
def predict(img):
|
| 10 |
+
img = load_image(img)
|
| 11 |
category, idx, probs = neural_net.predict(img)
|
| 12 |
return 'code has run'
|
| 13 |
|