SickBoy commited on
Commit
16644be
1 Parent(s): c212b2f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -31,9 +31,9 @@ image1 = example["image"]
31
  words = example["tokens"]
32
  boxes = example["bboxes"]
33
  labels = example["ner_tags"]
34
- Image.open(dataset[2]["image_path"]).convert("RGB").save("example1.png")
35
- Image.open(dataset[1]["image_path"]).convert("RGB").save("example2.png")
36
- Image.open(dataset[0]["image_path"]).convert("RGB").save("example3.png")
37
  # define id2label, label2color
38
  #labels = dataset.features['ner_tags'].feature.names
39
  labels = ['O', 'HEADER', 'SUBHEADER', 'TEXTO', 'NUMERAL', 'RESUMEN']
 
31
  words = example["tokens"]
32
  boxes = example["bboxes"]
33
  labels = example["ner_tags"]
34
+ #Image.open(dataset[2]["image_path"]).convert("RGB").save("example1.png")
35
+ #Image.open(dataset[1]["image_path"]).convert("RGB").save("example2.png")
36
+ #Image.open(dataset[0]["image_path"]).convert("RGB").save("example3.png")
37
  # define id2label, label2color
38
  #labels = dataset.features['ner_tags'].feature.names
39
  labels = ['O', 'HEADER', 'SUBHEADER', 'TEXTO', 'NUMERAL', 'RESUMEN']