Spaces:
Sleeping
Sleeping
Commit
·
c343652
1
Parent(s):
08d7e1f
Update app.py
Browse files
app.py
CHANGED
@@ -36,11 +36,7 @@ with open("list.dat", 'rb') as f:
|
|
36 |
model = load_learner("model.pkl")
|
37 |
|
38 |
def predict(img):
|
39 |
-
logging.warning(f"{
|
40 |
-
img = ToTensor()(img)
|
41 |
-
img.permute(2, 0, 1)
|
42 |
-
img = img.reshape((1, *(img.shape))).type(torch.uint8)
|
43 |
-
logging.warning(f"{img.shape}")
|
44 |
pred, idx, probs = model.predict(img)
|
45 |
logging.debug(f"{type(probs)}")
|
46 |
logging.debug(f"{type(probs[0])}")
|
|
|
36 |
model = load_learner("model.pkl")
|
37 |
|
38 |
def predict(img):
|
39 |
+
logging.warning(f"{img}")
|
|
|
|
|
|
|
|
|
40 |
pred, idx, probs = model.predict(img)
|
41 |
logging.debug(f"{type(probs)}")
|
42 |
logging.debug(f"{type(probs[0])}")
|