Update app.py

app.py

@@ -47,9 +47,9 @@ elif source_radio == "VIDEO":
 
 elif source_radio == "GAME":
     input_img = st.sidebar.file_uploader("이미지 파일을 선택하세요.", type=("jpg", "png"))
-    col1, col2 = st.columns([3, 1])  # Adjust the column widths as needed
     if input_img is None:
-        st.session_state.random_class = object_detection.call_class()
+        st.session_state.random_class = object_detection.call_class() 
+        col1, col2 = st.columns([3, 1])  # Adjust the column widths as needed
     
         with col1:
             st.title(f':{st.session_state.random_class}: 있는 이미지를 업로드')  # Display the header
@@ -57,20 +57,13 @@ elif source_radio == "GAME":
         with col2:
             if st.button('새로운 객체'):
                 st.session_state.random_class = object_detection.call_class()
-                 
     else:
         result_img, detected_object = object_detection.process_image(input_img)
-
-        with col1:
-               
-            if st.session_state.random_class  in detected_object:
-                st.header(f':{st.session_state.random_class}: 찾습니다!')
+        if st.session_state.random_class  in detected_object:
+            st.header(f':{st.session_state.random_class}: 찾습니다!')
             
-            else:
-                st.header(f':{st.session_state.random_class}: 찾을 수 없습니다')
-            st.image(result_img) 
-        
-        with col2:
-            if st.button('새로운 객체'):
-                st.session_state.random_class = object_detection.call_class()
+        else:
+            st.header(f':{st.session_state.random_class}: 찾을 수 없습니다')
+
+        st.image(result_img)