Hugging Face's logo

Spaces:
brainai-projects
/
object_detection_Streamlit
Sleeping

App Files Community
BrainAI-1 commited on
Commit
58fa179
·
verified ·
1 Parent(s): 26e8322

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -54,8 +54,9 @@ elif source_radio == "GAME":
54
  st.session_state.random_class = object_detection.call_class()
55
  else:
56
  result_img, detected_object = object_detection.process_image(input_img)
57
- detected_objects = detected_object.split(":")[1].strip()
58
- detected_objects_list = detected_object.split(", ")
 
59
  st.write(detected_objects_list)
60
 
61
  if st.session_state.random_class in detected_objects_list:
 
54
  st.session_state.random_class = object_detection.call_class()
55
  else:
56
  result_img, detected_object = object_detection.process_image(input_img)
57
+ detected_objects_list = detected_string.split(": ")[1].split(", ")
58
+ #detected_objects = detected_object.split(":")[1].strip()
59
+ #detected_objects_list = detected_object.split(", ")
60
  st.write(detected_objects_list)
61
 
62
  if st.session_state.random_class in detected_objects_list: