Update app.py

app.py

@@ -1,6 +1,7 @@
 #run the app
 #python -m streamlit run d:/NSFW/Project/test1.py
 import torch
+import tensorflow
 from transformers import BertTokenizer, BertForSequenceClassification
 import math, keras_ocr
 # Initialize pipeline
@@ -79,6 +80,7 @@ if uploaded_file is not None:
 # prediction_groups is a list of (word, box) tuples
     prediction_groups = pipeline.recognize([read_image])
     predictions = prediction_groups[0] # extract text list
+    tf.keras.backend.clear_session()
     predictions = get_distance(predictions)
     # Set thresh higher for text further apart
     predictions = list(distinguish_rows(predictions, thresh=10))