Update app.py

app.py

@@ -55,19 +55,16 @@ elif source_radio == "GAME":
     else:
         result_img, detected_object = object_detection.process_image(input_img)
         detected_objects_list = detected_object.split(", ")  
-        
-
 
+        if st.session_state.random_class  in detected_objects_list:
+            st.header(f':{st.session_state.random_class}: 찾습니다!')
+            
+        elif st.session_state.random_class not in detected_object:
+            st.header(f':{st.session_state.random_class}: 찾을 수 없습니다')
+ 
 
-        with col1:
-            if st.session_state.random_class  in detected_objects_list:
-                st.header(f':{st.session_state.random_class}: 찾습니다!')
-                
-            elif st.session_state.random_class not in detected_object:
-                st.header(f':{st.session_state.random_class}: 찾을 수 없습니다')
-        with col2:
-            if st.button('새로운 객체'):
-                st.session_state.random_class = object_detection.call_class()
         st.image(result_img)    
+        if st.button('새로운 객체'):
+            st.session_state.random_class = object_detection.call_class()