Update app.py
Browse files
app.py
CHANGED
@@ -21,15 +21,15 @@ with col1:
|
|
21 |
if model_choice == "EDSR":
|
22 |
|
23 |
model = EdsrModel.from_pretrained("eugenesiow/edsr-base", scale=2)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
24 |
else:
|
25 |
st.text('Not supported yet. Comeback soon for fun releases')
|
26 |
-
|
27 |
-
inputs = ImageLoader.load_image(image)
|
28 |
-
preds = model(inputs)
|
29 |
-
|
30 |
-
st.subheader('New image')
|
31 |
-
st.image(preds, cpation=f'Upscaled {choice} times image')
|
32 |
-
st.download_button('Download', preds)
|
33 |
|
34 |
|
35 |
|
|
|
21 |
if model_choice == "EDSR":
|
22 |
|
23 |
model = EdsrModel.from_pretrained("eugenesiow/edsr-base", scale=2)
|
24 |
+
inputs = ImageLoader.load_image(image)
|
25 |
+
preds = model(inputs)
|
26 |
+
|
27 |
+
st.subheader('New image')
|
28 |
+
st.image(preds, cpation=f'Upscaled {choice} times image')
|
29 |
+
st.download_button('Download', preds)
|
30 |
+
|
31 |
else:
|
32 |
st.text('Not supported yet. Comeback soon for fun releases')
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
33 |
|
34 |
|
35 |
|