Update app.py

app.py

@@ -24,29 +24,19 @@ if file_video is not None:
     sc = smartcities()
     output = sc.predict(file_video)
     
-    # col1, col2 = st.columns(2)    
+    col1, col2 = st.columns(2)    
     if output is not None:
-        # with col1:
-        #     st.subheader("Input: ")
-        #     video = open(file_video, "rb")
-        #     video_bytes = video.read()
-        #     st.video(video_bytes, format="video/mp4")
-        # with col2:
-        #     st.subheader("Output: ")
-        #     output_video = open(output, "rb")
-        #     output_bytes = output_video.read()
-        #     st.video(output_bytes, format="video/mp4")
-        #     st.download_button("Download", output_bytes, file_name="output_video.mp4", mime="video/mp4")   
-
-        st.subheader("Input: ")
-        video = open(file_video, "rb")
-        video_bytes = video.read()
-        st.video(video_bytes, format="video/mp4")
-
-        st.subheader("Output: ")
-        output_video = open(output, "rb")
-        output_bytes = output_video.read()
-        st.video(output_bytes, format="video/mp4")
-        st.download_button("Download", output_bytes, file_name="output_video.mp4", mime="video/mp4")   
+        with col1:
+            st.subheader("Input: ")
+            video = open(file_video, "rb")
+            video_bytes = video.read()
+            st.video(video_bytes, format="video/mp4")
+        with col2:
+            st.subheader("Output: ")
+            output_video = open(output, "rb")
+            output_bytes = output_video.read()
+            st.video(output_bytes, format="video/mp4")
+            st.download_button("Download", output_bytes, file_name="output_video.mp4", mime="video/mp4")   
+