Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -163,7 +163,8 @@ def run_detr(image:Image, detr_processor, detr_model, confidence_threshold: floa
|
|
163 |
outputs = postprocessed_outputs[0]
|
164 |
# dict{scores, logits, labels, boxes}
|
165 |
print(f"Output dict is {outputs.keys()}: {outputs}")
|
166 |
-
|
|
|
167 |
labels = [detr_model.config.id2label[label.item()] for label in outputs["labels"]]
|
168 |
boxes = outputs['boxes']
|
169 |
return boxes, labels
|
|
|
163 |
outputs = postprocessed_outputs[0]
|
164 |
# dict{scores, logits, labels, boxes}
|
165 |
print(f"Output dict is {outputs.keys()}: {outputs}")
|
166 |
+
ff = outputs['labels']
|
167 |
+
print(f"Labels are {ff}")
|
168 |
labels = [detr_model.config.id2label[label.item()] for label in outputs["labels"]]
|
169 |
boxes = outputs['boxes']
|
170 |
return boxes, labels
|