SalmanHabeeb commited on
Commit
c343652
·
1 Parent(s): 08d7e1f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -5
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"{type(img)}")
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])}")