| import streamlit as st | |
| from super_image import EdsrModel, ImageLoader | |
| from PIL import Image | |
| import cv2 | |
| 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", scale=2) | |
| inputs = ImageLoader.load_image(image) | |
| preds = model(inputs) | |
| ImageLoader.save_image(preds, './scaled_2x.png') | |
| img = cv2.imread('./scaled_2x.png') | |
| st.subheader('New image') | |
| st.image(img, caption=f'Upscaled {choice} times image') | |
| st.download_button('Download', img) | |
| else: | |
| st.text('Not supported yet. Comeback soon for fun releases') | |