SalmanHabeeb commited on
Commit
7bed125
·
1 Parent(s): d28f32d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -38,9 +38,9 @@ model = load_learner("model.pkl")
38
  def predict(img):
39
  logging.warning(f"{img}")
40
  try:
41
- logger.warning(f"Shape: {img.shape}")
42
  except AttributeError:
43
- logger.warning(f"Size: {img.size}")
44
  pred, idx, probs = model.predict(img)
45
  logging.debug(f"{type(probs)}")
46
  logging.debug(f"{type(probs[0])}")
 
38
  def predict(img):
39
  logging.warning(f"{img}")
40
  try:
41
+ logging.warning(f"Shape: {img.shape}")
42
  except AttributeError:
43
+ logging.warning(f"Size: {img.size}")
44
  pred, idx, probs = model.predict(img)
45
  logging.debug(f"{type(probs)}")
46
  logging.debug(f"{type(probs[0])}")