here

app.py

@@ -47,12 +47,31 @@ def train_and_inference(api_key, ontology_id, model_run_id):
     st.write('zat')
     from labelbox import Client
     st.write('8')
-    from labelbox.data import (
-        Label, ImageData,
-        Radio,
+    st.write('9')
+    import numpy as np
+    import tensorflow as tf
+    from tensorflow.keras import layers
+    from tensorflow.keras.models import Sequential
+    from tensorflow.keras.preprocessing.image import ImageDataGenerator
+    import os
+    from labelbox.schema.ontology import OntologyBuilder, Tool, Classification, Option
+    from labelbox import Client, LabelingFrontend, LabelImport, MALPredictionImport
+    from labelbox.data.annotation_types import (
+        Label, ImageData, ObjectAnnotation, MaskData,
+        Rectangle, Point, Line, Mask, Polygon,
+        Radio, Checklist, Text,
         ClassificationAnnotation, ClassificationAnswer
     )
-    st.write('9')
+    from labelbox import MediaType
+    from labelbox.data.serialization import NDJsonConverter
+    import pandas as pd
+    import shutil
+    import labelbox.data
+
+    import json
+    import uuid
+    import time
+    import requests
     import pandas as pd
     import shutil
 

requirements.txt

@@ -1,5 +1,5 @@
 streamlit
 numpy
 pandas
-labelbox
+labelbox[data]
 tensorflow