Update app.py
Browse files
app.py
CHANGED
@@ -66,7 +66,7 @@ def predict(image):
|
|
66 |
# Format the result string
|
67 |
result = f"Top Predictions:\n"
|
68 |
for i in range(top_prob.size(0)):
|
69 |
-
result += f"{classes[top_class[i]]}: {top_prob[i].item()
|
70 |
result += f"Prediction Time: {prediction_time:.2f} seconds"
|
71 |
|
72 |
return result
|
|
|
66 |
# Format the result string
|
67 |
result = f"Top Predictions:\n"
|
68 |
for i in range(top_prob.size(0)):
|
69 |
+
result += f"{classes[top_class[i]]}: Score {top_prob[i].item()}\n"
|
70 |
result += f"Prediction Time: {prediction_time:.2f} seconds"
|
71 |
|
72 |
return result
|