Update app.py

app.py

@@ -21,15 +21,15 @@ with col1:
         if model_choice == "EDSR":
 
             model = EdsrModel.from_pretrained("eugenesiow/edsr-base", scale=2)
+            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)
+            
         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)