Update prediction.py

prediction.py

@@ -3,12 +3,14 @@ import numpy as np
 from PIL import Image
 import tensorflow as tf
 from tensorflow.keras.models import load_model
-from tensorflow.keras.layers import KerasLayer
+import tensorflow_hub as hub
+
+
 
 # Load model sekali saat aplikasi di-start
 @st.cache_resource
 def load_skin_model():
-    return load_model('model_aug.keras', custom_objects={'KerasLayer': KerasLayer})
+    return load_model('model_aug.keras', custom_objects={'KerasLayer': hub.KerasLayer})
 
 model = load_skin_model()