Widium commited on
Commit
07ed221
·
1 Parent(s): d2cf515

remove annotation inside predict function

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -44,7 +44,7 @@ model.to(device="cpu")
44
 
45
  # ============================================================================= #
46
 
47
- def predict(img : Image)-> Tuple[Dict, float]:
48
  """
49
  Predict the class probabilities for a given image using a Pytorch Module,
50
  and return the predictions dictionary with the inference time.
 
44
 
45
  # ============================================================================= #
46
 
47
+ def predict(img)-> Tuple[Dict, float]:
48
  """
49
  Predict the class probabilities for a given image using a Pytorch Module,
50
  and return the predictions dictionary with the inference time.