analist commited on
Commit
3a11649
1 Parent(s): 4c79488

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +10 -7
app.py CHANGED
@@ -17,7 +17,10 @@ with col1:
17
  file = st.file_uploader('Post your picture')
18
  if file is not None:
19
  image = Image.open(file)
20
- st.image(image)
 
 
 
21
 
22
  if model_choice == "EDSR":
23
 
@@ -27,12 +30,12 @@ with col1:
27
 
28
  ImageLoader.save_image(preds, './scaled_2x.png')
29
  img = Image.open('./scaled_2x.png')
30
-
31
- st.subheader('New image')
32
- st.image(img, caption=f'Upscaled {choice} times image')
33
-
34
- with open('./scaled_2x.png', 'rb') as f:
35
- st.download_button(label="Download image", data=f, file_name="upscaled.png", mime="image/png")
36
 
37
  else:
38
  st.text('Not supported yet. Comeback soon for fun releases')
 
17
  file = st.file_uploader('Post your picture')
18
  if file is not None:
19
  image = Image.open(file)
20
+
21
+ col1, col2 = st.columns(2)
22
+ with col1:
23
+ st.image(image)
24
 
25
  if model_choice == "EDSR":
26
 
 
30
 
31
  ImageLoader.save_image(preds, './scaled_2x.png')
32
  img = Image.open('./scaled_2x.png')
33
+ with col2:
34
+ st.subheader('New image')
35
+ st.image(img, caption=f'Upscaled {choice} times image')
36
+
37
+ with open('./scaled_2x.png', 'rb') as f:
38
+ st.download_button(label="Download image", data=f, file_name="upscaled.png", mime="image/png")
39
 
40
  else:
41
  st.text('Not supported yet. Comeback soon for fun releases')