chmcbs commited on
Commit
3359f8d
·
verified ·
1 Parent(s): 5e5f960

Improve formatting of head count message

Browse files
Files changed (1) hide show
  1. 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...")