Update app.py

app.py

@@ -1,5 +1,7 @@
 import pandas_profiling as pp
 import pandas as pd
+import tensorflow as tf
+
 from datasets import load_dataset
 from tensorflow.python.framework import tensor_shape
 
@@ -135,7 +137,8 @@ def fn(    text1,    text2,    num,    slider1,    slider2,    single_checkbox,
     tokenizer = AutoTokenizer.from_pretrained(model_ckpt)
     model = TFAutoModel.from_pretrained(model_ckpt, from_pt=True)
     
-    TensorShape([1, 768])
+#    TensorShape([1, 768])
+    tf.shape([1, 768])
     
     embeddings_dataset = clinical_dataset.map(
     lambda x: {"embeddings": get_embeddings(x["text"]).numpy()[0]})