Update app.py
Browse files
app.py
CHANGED
@@ -20,11 +20,12 @@ with col1:
|
|
20 |
|
21 |
col1, col2 = st.columns(2)
|
22 |
with col1:
|
|
|
23 |
st.image(image)
|
24 |
|
25 |
if model_choice == "EDSR":
|
26 |
|
27 |
-
model = EdsrModel.from_pretrained("eugenesiow/edsr-base", scale=
|
28 |
inputs = ImageLoader.load_image(image)
|
29 |
preds = model(inputs)
|
30 |
|
|
|
20 |
|
21 |
col1, col2 = st.columns(2)
|
22 |
with col1:
|
23 |
+
st.subheader('Input image')
|
24 |
st.image(image)
|
25 |
|
26 |
if model_choice == "EDSR":
|
27 |
|
28 |
+
model = EdsrModel.from_pretrained("eugenesiow/edsr-base", scale=int(choice))
|
29 |
inputs = ImageLoader.load_image(image)
|
30 |
preds = model(inputs)
|
31 |
|