Update app.py

app.py

@@ -11,10 +11,8 @@ st.set_page_config(
 st.title(":blue[Object Detection] :black_cat:")
 st.sidebar.header("메뉴")
 source_radio = st.sidebar.radio("선택하세요", ["IMAGE", "VIDEO", "GAME"])
-st.session_state.random_class = None
-if not st.session_state.random_class:
 
-    st.session_state.random_class = object_detection.call_class()
+    
 
 
 if source_radio == "IMAGE":
@@ -51,6 +49,7 @@ 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()
     
         with col1:
             st.title(f':{st.session_state.random_class}: 있는 이미지를 업로드')  # Display the header