Update app.py
Browse files
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 |
-
|
| 59 |
-
if st.session_state.random_class in
|
| 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:
|