Hugging Face's logo

Spaces:
codewithRiz
/
Buck_Tracker
Sleeping

App Files Community
codewithRiz commited on
Commit
25432a2
·
1 Parent(s): d72629b

retagging images

Browse files
Files changed (1) hide show
  1. api/detection.py +5 -5
api/detection.py CHANGED
@@ -80,15 +80,15 @@ async def predict(
80
  # Stage 2: Doe
81
  # ================================================================
82
  VALID_LABELS = {
83
- "Deer | Doe", # trailing space matches pipeline output
84
- "Deer | Buck | White Tail Bucks",
85
- "Deer | Buck | Mule Bucks",
86
  }
87
 
88
  VALID_LABELS_DISPLAY = [ # clean version shown in error messages
89
  "Deer | Doe",
90
- "Deer | Buck | White Tail Bucks",
91
- "Deer | Buck | Mule Bucks",
92
 
93
 
94
  ]
 
80
  # Stage 2: Doe
81
  # ================================================================
82
  VALID_LABELS = {
83
+ "Deer | Doe ", # trailing space matches pipeline output
84
+ "Deer | Buck | Whitetail",
85
+ "Deer | Buck | Mule",
86
  }
87
 
88
  VALID_LABELS_DISPLAY = [ # clean version shown in error messages
89
  "Deer | Doe",
90
+ "Deer | Buck | Whitetail",
91
+ "Deer | Buck | Mule",
92
 
93
 
94
  ]