karthickn commited on
Commit
fae9073
·
1 Parent(s): 0a5cba8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +0 -1
app.py CHANGED
@@ -53,7 +53,6 @@ def process_image(image, margin, model):
53
  draw.rectangle([(box[0], box[1]), (box[2], box[3])], outline=colors[index], width=4)
54
  gallery.append(image.crop((box[0], box[1], box[2], box[3])))
55
  labels.append(model.config.id2label[label.item()])
56
- print(model.config.id2label[label.item()])
57
  newlabel[model.config.id2label[label.item()]] = 1
58
  index += 1
59
  yield [drawImage, gallery, newlabel, ','.join(labels)]
 
53
  draw.rectangle([(box[0], box[1]), (box[2], box[3])], outline=colors[index], width=4)
54
  gallery.append(image.crop((box[0], box[1], box[2], box[3])))
55
  labels.append(model.config.id2label[label.item()])
 
56
  newlabel[model.config.id2label[label.item()]] = 1
57
  index += 1
58
  yield [drawImage, gallery, newlabel, ','.join(labels)]