Spaces:
Runtime error
Runtime error
Widium
commited on
Commit
·
07ed221
1
Parent(s):
d2cf515
remove annotation inside predict function
Browse files
app.py
CHANGED
|
@@ -44,7 +44,7 @@ model.to(device="cpu")
|
|
| 44 |
|
| 45 |
# ============================================================================= #
|
| 46 |
|
| 47 |
-
def predict(img
|
| 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.
|