Update app.py

app.py

@@ -11,9 +11,9 @@ st.set_page_config(
 st.title(":blue[Object Detection] :black_cat:")
 st.sidebar.header("메뉴")
 source_radio = st.sidebar.radio("선택하세요", ["IMAGE", "VIDEO", "GAME"])
-if 'random_class' not in st.session_state:
-    st.session_state.random_class = object_detection.call_class()
-    st.write('initialize')
+
+st.session_state.random_class = None
+
 
 if source_radio == "IMAGE":
     st.write(":green[왼쪽 메뉴 'Browse files' 버튼을 클릭하여 이미지 파일을 선택하면 AI 추론이 시작됩니다.]" )
@@ -49,7 +49,6 @@ 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:
-        
     
         with col1:
             st.title(f':{st.session_state.random_class}: 있는 이미지를 업로드')  # Display the header
@@ -57,7 +56,7 @@ elif source_radio == "GAME":
         with col2:
             if st.button('새로운 객체'):
                 st.session_state.random_class = object_detection.call_class()
-                st.write("new object 1")
+                 
     else:
         result_img, detected_object = object_detection.process_image(input_img)
 
@@ -73,4 +72,4 @@ elif source_radio == "GAME":
         with col2:
             if st.button('새로운 객체'):
                 st.session_state.random_class = object_detection.call_class()
-                st.write("new object2")
+