Spaces:
Runtime error
Runtime error
setting default model to mobilenet_v2
Browse files
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:
|