File size: 809 Bytes
70d7fb2 2ec19ea 70d7fb2 6dd0277 70d7fb2 2ec19ea 485af7c 2ec19ea 6dd0277 70d7fb2 6228942 70d7fb2 2ec19ea 6dd0277 2ec19ea 6dd0277 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 |
import streamlit as st
from super_image import EdsrModel, ImageLoader
from PIL import Image
st.title('Take your picture to the next level')
col1, col2 = st.columns([3, 1])
preds = None
with col1:
file = st.file_uploader('Post your picture')
if file is not None:
image = Image.open(file)
st.image(image)
with col2:
choice = st.selectbox('Scale to', ['x2', 'x3', 'x4'])
model_choice = st.selectbox('Model', ['EDSR', 'PAN', 'CARN'])
st.text(choice)
if model_choice == "EDSR":
model = EdsrModel.from_pretrained('eugenesiow/edsr-base', scale=choice)
else:
st.text('Not supported yet. Comeback soon for fun releases')
inputs = ImageLoader.load_image(image)
preds = model(inputs)
st.subheader('New image')
st.image(preds, cpation=f'Upscaled {choice} times image') |