Update app.py

app.py

@@ -24,15 +24,15 @@ if file_video is not None:
     sc = smartcities()
     output = sc.predict(file_video)
     
-    raw1 = st.columns(1)    
-    raw2 = st.columns(1)
+    # raw1 = st.columns(1)    
+    # raw2 = st.columns(1)
     if output is not None:
-        with raw1:
+        with st.container():
             st.subheader("Input: ")
             video = open(file_video, "rb")
             video_bytes = video.read()
             st.video(video_bytes, format="video/mp4")
-        with raw2:
+        with st.container():
             st.subheader("Output: ")
             # output_video = open(output, "rb")
             # output_bytes = output_video.read()