Update app.py
Browse files
app.py
CHANGED
|
@@ -23,9 +23,11 @@ with col1:
|
|
| 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,
|
| 29 |
st.download_button('Download', preds)
|
| 30 |
|
| 31 |
else:
|
|
|
|
| 23 |
model = EdsrModel.from_pretrained("eugenesiow/edsr-base", scale=2)
|
| 24 |
inputs = ImageLoader.load_image(image)
|
| 25 |
preds = model(inputs)
|
| 26 |
+
|
| 27 |
+
for i in range(101):
|
| 28 |
+
st.progress(i, text='Magic is happening')
|
| 29 |
st.subheader('New image')
|
| 30 |
+
st.image(preds, caption=f'Upscaled {choice} times image')
|
| 31 |
st.download_button('Download', preds)
|
| 32 |
|
| 33 |
else:
|