Update app.py

app.py

@@ -17,7 +17,10 @@ with col1:
     file = st.file_uploader('Post your picture')
     if file is not None:
         image = Image.open(file)
-        st.image(image)
+
+        col1, col2 = st.columns(2)
+        with col1:
+            st.image(image)
 
         if model_choice == "EDSR":
 
@@ -27,12 +30,12 @@ with col1:
 
             ImageLoader.save_image(preds, './scaled_2x.png')
             img = Image.open('./scaled_2x.png')
-            
-            st.subheader('New image')
-            st.image(img, caption=f'Upscaled {choice} times image')
-
-            with open('./scaled_2x.png', 'rb') as f:
-                st.download_button(label="Download image", data=f, file_name="upscaled.png", mime="image/png")
+            with col2:
+                st.subheader('New image')
+                st.image(img, caption=f'Upscaled {choice} times image')
+    
+                with open('./scaled_2x.png', 'rb') as f:
+                    st.download_button(label="Download image", data=f, file_name="upscaled.png", mime="image/png")
             
         else:
             st.text('Not supported yet. Comeback soon for fun releases')