x

app.py

@@ -29,6 +29,7 @@ example_idx = int(col1.number_input('Sample an example from the validation set',
 img_set, idx, descr = descriptions[example_idx]
 idx = int(idx)
 images = [prefix+'/'+img_set+'/'+i for i in set2ids[img_set]]
+img_urls = images.copy()
 index = int(col2.number_input('Image Index from 0 to 9', value=0, min_value=0, max_value=len(images)-1))
 
 col1.subheader(descr)
@@ -43,12 +44,12 @@ if show:
     if index == idx:
         cap = f'{index} (GROUNDTRUTH)'
     caps[idx] = f'{idx} (GROUNDTRUTH)'
-    images[idx] = ImageOps.expand(Image.open(io.BytesIO(requests.get(images[index], stream=True).content)),border=50,fill='green')
+    images[idx] = ImageOps.expand(Image.open(io.BytesIO(requests.get(img_urls[index], stream=True).content)),border=50,fill='green')
 if hide:
     if index == idx:
         cap = f'{index}'
     caps[idx] = f'{idx}'
-    images[idx] = Image.open(io.BytesIO(requests.get(images[index], stream=True).content))
+    images[idx] = Image.open(io.BytesIO(requests.get(img_urls[index], stream=True).content))
 # shown = True if len(caps[idx]) > 2 else False
 # if shown 
 col1.image(img, use_column_width=True, caption=cap)