debug deployment
Browse files- .gitattributes +2 -3
- app.py +2 -0
.gitattributes
CHANGED
|
@@ -1,3 +1,2 @@
|
|
| 1 |
-
*.
|
| 2 |
-
|
| 3 |
-
panda-model-1.pkl filter=lfs diff=lfs merge=lfs -text
|
|
|
|
| 1 |
+
*.pkl filter=lfs diff=lfs merge=lfs -text
|
| 2 |
+
*.jpg filter=lfs diff=lfs merge=lfs -text
|
|
|
app.py
CHANGED
|
@@ -8,6 +8,8 @@ learn = load_learner('panda-model-1.pkl')
|
|
| 8 |
labels = learn.dls.vocab
|
| 9 |
|
| 10 |
def predict(img):
|
|
|
|
|
|
|
| 11 |
img = get_crops(PILImage.create(img))
|
| 12 |
pred,pred_idx,probs = learn.predict(img)
|
| 13 |
return {labels[i]: float(probs[i]) for i in range(len(labels))}
|
|
|
|
| 8 |
labels = learn.dls.vocab
|
| 9 |
|
| 10 |
def predict(img):
|
| 11 |
+
print(type(img))
|
| 12 |
+
print(img.shape)
|
| 13 |
img = get_crops(PILImage.create(img))
|
| 14 |
pred,pred_idx,probs = learn.predict(img)
|
| 15 |
return {labels[i]: float(probs[i]) for i in range(len(labels))}
|