Update app.py

app.py

@@ -12,9 +12,6 @@ st.title(":blue[Object Detection] :black_cat:")
 st.sidebar.header("메뉴")
 source_radio = st.sidebar.radio("선택하세요", ["IMAGE", "VIDEO", "GAME"])
 
-    
-
-
 if source_radio == "IMAGE":
     st.write(":green[왼쪽 메뉴 'Browse files' 버튼을 클릭하여 이미지 파일을 선택하면 AI 추론이 시작됩니다.]" )
     st.sidebar.header("이미지 파일 업로드")
@@ -38,22 +35,21 @@ elif source_radio == "VIDEO":
     st.write(":green[왼쪽 메뉴 'Browse files' 버튼을 클릭하여 비디오 파일을 선택하면 AI 추론이 시작됩니다.]" )
     st.sidebar.header("비디오 파일 업로드")
     input_video = st.sidebar.file_uploader("비디오 파일을 선택하세요..", type=("mp4"))
+    print(input_video)
     if input_video is not None:
         temp_file = object_detection.play_video(input_video)
         st.video(temp_file)
     else:
         st.video("data/breakfast.mp4")
 
-
 elif source_radio == "GAME":
     input_img = st.sidebar.file_uploader("이미지 파일을 선택하세요.", type=("jpg", "png"))
     if input_img is None:
         st.session_state.random_class = object_detection.call_class() 
-        col1, col2 = st.columns([3, 1])  # Adjust the column widths as needed
+        col1, col2 = st.columns([3, 1])  
     
         with col1:
-            st.title(f':{st.session_state.random_class}: 있는 이미지를 업로드')  # Display the header
-        
+            st.title(f':{st.session_state.random_class}: 있는 이미지를 업로드')  
         with col2:
             if st.button('새로운 객체'):
                 st.session_state.random_class = object_detection.call_class()
@@ -66,4 +62,5 @@ elif source_radio == "GAME":
             st.header(f':{st.session_state.random_class}: 찾을 수 없습니다')
 
         st.image(result_img)    
-                 
+   
+