rajistics commited on
Commit
06faa9e
β€’
1 Parent(s): 82536fc

baby steps

Browse files
Files changed (1) hide show
  1. app.py +4 -2
app.py CHANGED
@@ -75,8 +75,10 @@ def process_image(image):
75
  font = ImageFont.load_default()
76
  for prediction, box in zip(true_predictions, true_boxes):
77
  predicted_label = iob_to_label(prediction).lower()
78
- draw.rectangle(box, outline=label2color[predicted_label])
79
- draw.text((box[0]+10, box[1]-10), text=predicted_label, fill=label2color[predicted_label], font=font)
 
 
80
 
81
  return image
82
 
 
75
  font = ImageFont.load_default()
76
  for prediction, box in zip(true_predictions, true_boxes):
77
  predicted_label = iob_to_label(prediction).lower()
78
+ print (predicted_label)
79
+ print (label2color[predicted_label])
80
+ #draw.rectangle(box, outline=label2color[predicted_label])
81
+ #draw.text((box[0]+10, box[1]-10), text=predicted_label, fill=label2color[predicted_label], font=font)
82
 
83
  return image
84