Update app.py

app.py

@@ -151,32 +151,32 @@ with col2:
 
     _, dcol, _ = st.columns([1,5,1],gap="small")
 
-    with dcol:
+    
 
-        if st.button("PROCEED",use_container_width=True) and file is not None:
-            if st.session_state.get('opt1') == True:
-                task = st.session_state.opt1_selBox
-            else:
-                task = [st.session_state.width, st.session_state.height]
-            print(task)
-            st.session_state.disable_download = not upscale(file,task)
+    if st.button("PROCEED",use_container_width=True) and file is not None:
+        if st.session_state.get('opt1') == True:
+            task = st.session_state.opt1_selBox
+        else:
+            task = [st.session_state.width, st.session_state.height]
+        print(task)
+        st.session_state.disable_download = not upscale(file,task)
 
 
-            #print(resulted_file.shape)
+        #print(resulted_file.shape)
 
-        st.markdown("\n")
-        st.markdown("\n")
+    st.markdown("\n")
+    st.markdown("\n")
 
-        if file is None:
-            st.session_state.disable_download = True
+    if file is None:
+        st.session_state.disable_download = True
 
-        if st.session_state.disable_download == True:
-            st.button("DOWNLOAD FILE",disabled=True,use_container_width=True)
-        else:
-            with open('processed_'+file.name, "rb") as download_file:
-                st.download_button(label="Download image", data=download_file,
-                                   file_name= 'processed_'+file.name, mime= "image/png",
-                                   use_container_width=True, disabled=st.session_state.disable_download)
+    if st.session_state.disable_download == True:
+        st.button("DOWNLOAD FILE",disabled=True,use_container_width=True)
+    else:
+        with open('processed_'+file.name, "rb") as download_file:
+            st.download_button(label="Download image", data=download_file,
+                                file_name= 'processed_'+file.name, mime= "image/png",
+                                use_container_width=True, disabled=st.session_state.disable_download)