Update app.py

app.py

@@ -5,13 +5,14 @@ from PIL import Image
 st.title('Take your picture to the next level')
 
 col1, col2 = st.columns([3, 1])
+preds = None
 
 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', ['x2', 'x3', 'x4'])
     model_choice = st.selectbox('Model', ['EDSR', 'PAN', 'CARN'])
@@ -19,10 +20,13 @@ with col2:
 st.text(choice)
 
 
-"""
-model = EdsrModel.from_pretrained('eugenesiow/edsr-base', scale=2)
+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)
 
-ImageLoader.save_image(preds, './scaled_2x.png')
-ImageLoader.save_compare(inputs, preds, './scaled_2x_compare.png')"""
+st.subheader('New image')
+st.image(preds, cpation=f'Upscaled {choice} times image')