Update safety_checker_improved.py

safety_checker_improved.py

@@ -9,11 +9,11 @@ import sys
 sys.path.append('/home/ryn_mote/Misc/generative_recommender/gradio_video/automl/efficientnetv2/')
 import tensorflow as tf
 from tensorflow.keras import mixed_precision
-physical_devices = tf.config.list_physical_devices('GPU')
+#physical_devices = tf.config.list_physical_devices('GPU')
 
-tf.config.experimental.set_memory_growth(
-    physical_devices[0], True
-)
+#tf.config.experimental.set_memory_growth(
+#    physical_devices[0], True
+#)
 
 model = tf.keras.models.load_model('nsfweffnetv2-b02-3epochs.h5',custom_objects={"KerasLayer":hub.KerasLayer})
 # "The image classifier had been trained on 682550 images from the 5 classes "Drawing" (39026), "Hentai" (28134), "Neutral" (369507), "Porn" (207969) & "Sexy" (37914).