Spaces:
Sleeping
Sleeping
Soham Chandratre
commited on
Commit
·
4dc6bee
1
Parent(s):
c9fb15b
minor changes
Browse files- model/pothole_model.py +2 -1
model/pothole_model.py
CHANGED
|
@@ -71,4 +71,5 @@ def load_image_model(image):
|
|
| 71 |
|
| 72 |
# Print prediction and confidence score
|
| 73 |
print("Class:", class_name[2:], end="")
|
| 74 |
-
print("Confidence Score:", confidence_score)
|
|
|
|
|
|
| 71 |
|
| 72 |
# Print prediction and confidence score
|
| 73 |
print("Class:", class_name[2:], end="")
|
| 74 |
+
print("Confidence Score:", confidence_score)
|
| 75 |
+
return class_name
|