Spaces:

arsath-sm
/

face_classification

Sleeping

arsath-sm commited on Jul 13

Commit

8b1b73a

•

1 Parent(s): 2423049

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -6,8 +6,8 @@ import numpy as np
 from huggingface_hub import from_pretrained_keras
 # Load the models
-model1 = from_pretrained_keras("arsath-sm/face_classification_model1")
-model2 = from_pretrained_keras("arsath-sm/face_classification_model2")
 # Preprocess the image
 def preprocess_image(img):

 from huggingface_hub import from_pretrained_keras
 # Load the models
+model1 = tf.keras.layers.TFSMLayer("arsath-sm/face_classification_model1", call_endpoint='serving_default')
+model2 = tf.keras.layers.TFSMLayer("arsath-sm/face_classification_model2", call_endpoint='serving_default')
 # Preprocess the image
 def preprocess_image(img):