Update app.py

app.py

@@ -20,9 +20,9 @@ vit_b16 = hub.KerasLayer(model_url, trainable=False,  input_shape=(224, 224, 3))
 # tf.config.experimental._could_use_gpu_kernel = False
 
 # model = TFAutoModelForSequenceClassification.from_pretrained(model_pb_path)
-model = tf.keras.models.load_model(model_pb_path)
+# model = tf.keras.models.load_model(model_pb_path)
 
-# model = tf.keras.models.load_model(model_keras_path, custom_objects={'KerasLayer': hub.KerasLayer})
+model = tf.keras.models.load_model(model_keras_path)
 # model = tf.keras.models.load_model(model_h5_path)
 # model.compile(optimizer=Adam(learning_rate=0.0001),
 #               loss='binary_crossentropy',