d22cs051 commited on
Commit
d6bb5f9
1 Parent(s): eed2ca3

setting default model to mobilenet_v2

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -39,7 +39,7 @@ model_name_to_path = {
39
 
40
 
41
  # Create predict function
42
- def predict(img, model_name: str,) -> Tuple[Dict, float]:
43
  """
44
  Desc: Transforms and performs a prediction on img and returns prediction and time taken.
45
  Args:
 
39
 
40
 
41
  # Create predict function
42
+ def predict(img, model_name: str = "mobilenet_v2",) -> Tuple[Dict, float]:
43
  """
44
  Desc: Transforms and performs a prediction on img and returns prediction and time taken.
45
  Args: