|
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 col2: |
|
choice = st.selectbox('Scale to', [2, 3, 4]) |
|
model_choice = st.selectbox('Model', ['EDSR', 'PAN', 'CARN']) |
|
|
|
with col1: |
|
file = st.file_uploader('Post your picture') |
|
if file is not None: |
|
image = Image.open(file) |
|
st.image(image) |
|
|
|
if model_choice == "EDSR": |
|
|
|
model = EdsrModel.from_pretrained("eugenesiow/edsr-base") |
|
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') |
|
st.download_button('Download', preds) |
|
|
|
|
|
|
|
|
|
|
|
|
|
|