Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
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)
|