skyBluezz commited on
Commit
789f911
·
verified ·
1 Parent(s): 860f0b2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
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
- print(f"Labels are {outputs["labels"]}")
 
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