Update app.py

app.py

@@ -20,11 +20,12 @@ with col1:
 
         col1, col2 = st.columns(2)
         with col1:
+            st.subheader('Input image')
             st.image(image)
 
         if model_choice == "EDSR":
 
-            model = EdsrModel.from_pretrained("eugenesiow/edsr-base", scale=2)
+            model = EdsrModel.from_pretrained("eugenesiow/edsr-base", scale=int(choice))
             inputs = ImageLoader.load_image(image)
             preds = model(inputs)