nihalbaig commited on
Commit
26f7d7e
1 Parent(s): 21fdde8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -25,7 +25,7 @@ model = AutoModelForTokenClassification.from_pretrained("Theivaprakasham/layoutl
25
 
26
  # load image example
27
  dataset = load_dataset("darentang/generated", split="test")
28
- Image.open("https://raw.githubusercontent.com/nihalbaig0/BD-Vehicle-Detection/main/images/bondor_to_kodomtoli.jpg").convert("RGB").save("example1.png")
29
  Image.open(dataset[1]["image_path"]).convert("RGB").save("example2.png")
30
  Image.open(dataset[0]["image_path"]).convert("RGB").save("example3.png")
31
  # define id2label, label2color
 
25
 
26
  # load image example
27
  dataset = load_dataset("darentang/generated", split="test")
28
+ Image.open(dataset[2]["image_path"]).convert("RGB").save("example1.png")
29
  Image.open(dataset[1]["image_path"]).convert("RGB").save("example2.png")
30
  Image.open(dataset[0]["image_path"]).convert("RGB").save("example3.png")
31
  # define id2label, label2color