Update app.py

app.py

@@ -4,35 +4,34 @@ from PIL import Image
 
 st.title('Take your picture to the next level')
 
-def upscale(model_choice):
-    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)
-    
 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)
 
-with col2:
-    choice = st.selectbox('Scale to', [2, 3, 4])
-    model_choice = st.selectbox('Model', ['EDSR', 'PAN', 'CARN'])
-    st.button('Upscale', on_click=upscale(model_choice))
+        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)
+
 
-st.text(choice)