Spaces:
Runtime error
Runtime error
from turtle import color, onclick | |
import streamlit as st | |
from PIL import Image, ImageOps | |
import glob | |
import json | |
import requests | |
import random | |
def sample(): | |
img_set = random.sample(list(set2ids.keys()),1)[0] | |
prefix = 'https://raw.githubusercontent.com/BennoKrojer/imagecode-val-set/main/image-sets-val/' | |
set2ids = json.load(open('set2ids.json', 'r')) | |
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']] | |
if st.button('Sample set of images with contextual description'): | |
img_set = random.sample(list(set2ids.keys()),1)[0] | |
index= int(col1.number_input('Image Index from 0 to 9', value=0, min_value=0, max_value=len(images)-1, color='red')) | |
img = images[index] | |
images[index] = ImageOps.expand(Image.open(requests.get(images[index], stream=True).raw),border=10,fill='blue') | |
col1.image(img, use_column_width=True) | |
col2.image(images, width=150, caption=list(range(10))) | |