BrainAI-1 commited on
Commit
f9ef5c5
·
verified ·
1 Parent(s): 203130a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -55,8 +55,8 @@ elif source_radio == "GAME":
55
  st.session_state.random_class = object_detection.call_class()
56
  else:
57
  result_img, detected_object = object_detection.process_image(input_img)
58
- st.write(detected_object)
59
- if st.session_state.random_class in detected_object:
60
  st.header(f':{st.session_state.random_class}: 찾습니다!')
61
 
62
  elif st.session_state.random_class not in detected_object:
 
55
  st.session_state.random_class = object_detection.call_class()
56
  else:
57
  result_img, detected_object = object_detection.process_image(input_img)
58
+ detected_objects_list = detected_object.split(", ")
59
+ if st.session_state.random_class in detected_objects_list:
60
  st.header(f':{st.session_state.random_class}: 찾습니다!')
61
 
62
  elif st.session_state.random_class not in detected_object: