x

app.py

@@ -20,7 +20,7 @@ img_set = random.sample(list(set2ids.keys()),1)[0]
 st.set_page_config(layout="wide")
 col1, col2 = st.columns(2)
 
-images = [prefix+'/'+img_set+'/'+i for i in set2ids['MSR-VTT-videoTestVideo_video7016-shot1_2']]
+images = [prefix+'/'+img_set+'/'+i for i in set2ids[img_set]]
 
 if st.button('Sample set of images with contextual description'):
     img_set = random.sample(list(set2ids.keys()),1)[0]    
@@ -40,9 +40,8 @@ index= int(col1.number_input('Image Index from 0 to 9', value=0, min_value=0, ma
 
 
 img = images[index]
-# images[index] = ImageOps.expand(Image.open(io.BytesIO(requests.get(images[index], stream=True).content)),border=10,fill='blue')
+images[index] = ImageOps.expand(Image.open(io.BytesIO(requests.get(images[index], stream=True).content)),border=10,fill='blue')
 
 col1.image(img, use_column_width=True)
 caps = list(range(10))
-caps[5] = "**5**"
 col2.image(images, width=150, caption=caps)