Improve formatting of head count message
Browse files- inference.py +1 -1
inference.py
CHANGED
|
@@ -155,7 +155,7 @@ if __name__ == "__main__":
|
|
| 155 |
|
| 156 |
# Count heads
|
| 157 |
num_heads = model.count_heads(predictions)
|
| 158 |
-
print(f"\n🌾 {num_heads} heads detected
|
| 159 |
|
| 160 |
# Create visualisations
|
| 161 |
print("\nGenerating visualisations...")
|
|
|
|
| 155 |
|
| 156 |
# Count heads
|
| 157 |
num_heads = model.count_heads(predictions)
|
| 158 |
+
print(f"\n🌾 {num_heads} heads detected")
|
| 159 |
|
| 160 |
# Create visualisations
|
| 161 |
print("\nGenerating visualisations...")
|