Spaces:
Runtime error
Runtime error
Update app.py
Browse files
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)]
|