Update app.py

app.py

@@ -7,6 +7,20 @@ from tensorflow.keras.layers.experimental.preprocessing import RandomHeight
 with CustomObjectScope({'RandomHeight': RandomHeight}):
     model_0 = tf.keras.models.load_model('bestmodel.h5')
 
+    
+print("TensorFlow version:", tf.__version__)
+print("GPU Available:", tf.config.list_physical_devices('GPU'))
+
+# Configure TensorFlow to use memory growth
+gpus = tf.config.experimental.list_physical_devices('GPU')
+if gpus:
+    try:
+        for gpu in gpus:
+            tf.config.experimental.set_memory_growth(gpu, True)
+    except RuntimeError as e:
+        print(e)
+
+
 def classify_image(inp):
     # Convert to PIL Image if we have a numpy array
     image = None