qubvel-hf HF staff commited on
Commit
0601eb3
·
verified ·
1 Parent(s): b53f046

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -43,7 +43,7 @@ def process_image(image: PIL.Image.Image) -> tuple[PIL.Image.Image, list[dict]]:
43
  indexes = result["labels"].cpu().numpy()
44
  scores = result["scores"].cpu().numpy()
45
  text_labels = [
46
- f"{model.id2label[index]} [{score.item():.2f}]" for index, score in zip(indexes, scores)
47
  ]
48
 
49
  detections = sv.Detections(xyxy=boxes_xyxy)
 
43
  indexes = result["labels"].cpu().numpy()
44
  scores = result["scores"].cpu().numpy()
45
  text_labels = [
46
+ f"{model.config.id2label[index]} [{score.item():.2f}]" for index, score in zip(indexes, scores)
47
  ]
48
 
49
  detections = sv.Detections(xyxy=boxes_xyxy)