Update app.py

app.py

@@ -4,6 +4,19 @@ 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', scale=choice)
+    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
 
@@ -16,18 +29,10 @@ with col1:
 with col2:
     choice = st.selectbox('Scale to', ['x2', 'x3', 'x4'])
     model_choice = st.selectbox('Model', ['EDSR', 'PAN', 'CARN'])
+    st.button('Upscale', on_click=upscale(model_choice))
 
 st.text(choice)
 
 
-if model_choice == "EDSR":
-    model = EdsrModel.from_pretrained('eugenesiow/edsr-base', scale=choice)
-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)
+