Update app.py

app.py

@@ -5,7 +5,9 @@ import numpy as np
 import tensorflow as tf
 import tensorflow_hub as hub
 from tensorflow.keras.models import load_model
+import os
 
+os.environ["CUDA_VISIBLE_DEVICES"] = "-1"
 # Specify the custom objects for loading the model
 custom_objects = {'KerasLayer': hub.KerasLayer}