Update app.py
Browse files
app.py
CHANGED
|
@@ -4,7 +4,7 @@ from fastai.vision.all import *
|
|
| 4 |
import skimage
|
| 5 |
|
| 6 |
learn=load_learner("model.pkl")
|
| 7 |
-
examples=[["
|
| 8 |
categories=('margherita pizza','pepperoni pizza','paneer pizza','hawaiian pizza','greeek pizza')
|
| 9 |
def classify_img(img):
|
| 10 |
pred,idx,probs=learn.predict(img)
|
|
|
|
| 4 |
import skimage
|
| 5 |
|
| 6 |
learn=load_learner("model.pkl")
|
| 7 |
+
examples=[["margherita.jpg"],["paneer.jpg"],["hawaiian.jpg"]]
|
| 8 |
categories=('margherita pizza','pepperoni pizza','paneer pizza','hawaiian pizza','greeek pizza')
|
| 9 |
def classify_img(img):
|
| 10 |
pred,idx,probs=learn.predict(img)
|