Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -36,10 +36,11 @@ learn = load_learner("templateClassifierDATAhalfEPOCHoneVISION.pkl")
|
|
| 36 |
print("Model Vocabulary:", learn.dls.vocab)
|
| 37 |
|
| 38 |
labels = learn.dls.vocab
|
| 39 |
-
|
| 40 |
-
|
| 41 |
-
|
| 42 |
-
|
|
|
|
| 43 |
|
| 44 |
def predict(img):
|
| 45 |
img = PILImage.create(img)
|
|
|
|
| 36 |
print("Model Vocabulary:", learn.dls.vocab)
|
| 37 |
|
| 38 |
labels = learn.dls.vocab
|
| 39 |
+
|
| 40 |
+
# def predict(img):
|
| 41 |
+
# img = PILImage.create(img)
|
| 42 |
+
# pred,pred_idx,probs = learn.predict(img)
|
| 43 |
+
# return {labels[i]: float(probs[i]) for i in range(len(labels))}
|
| 44 |
|
| 45 |
def predict(img):
|
| 46 |
img = PILImage.create(img)
|