Update app.py
Browse files
app.py
CHANGED
|
@@ -54,6 +54,7 @@ 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_list = detected_object.split(", ")
|
| 58 |
st.write(detected_objects_list)
|
| 59 |
|
|
|
|
| 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_string.split(":")[1].strip()
|
| 58 |
detected_objects_list = detected_object.split(", ")
|
| 59 |
st.write(detected_objects_list)
|
| 60 |
|