Update flask/Emotion_spotting_service.py

flask/Emotion_spotting_service.py

@@ -1,3 +1,4 @@
+import tensorflow as tf
 import tensorflow.keras as keras
 import numpy as np
 import librosa
@@ -5,6 +6,22 @@ import random
 #import tensorflow_addons
 SAMPLE_RATE = 22050
 
+import tensorflow as tf
+
+gpus = tf.config.list_physical_devices('GPU')
+if gpus:
+    try:
+        # Enable memory growth
+        for gpu in gpus:
+            tf.config.experimental.set_memory_growth(gpu, True)
+
+        # Optionally, set a memory limit
+        tf.config.set_logical_device_configuration(
+            gpus[0],
+            [tf.config.LogicalDeviceConfiguration(memory_limit=5120)])  # Set memory limit to 10GB
+    except RuntimeError as e:
+        print(e)
+
 class _Emotion_spotting_service():
     model = None
     #instance = None