Update app.py

app.py

@@ -114,7 +114,12 @@ def prediction(df):
   #print(X_test.shape)
   X_test_encoded = label_encoder(df)
   X_test_df = preprocess_dataset(X_test_encoded)
-  x_model = loaded_model = tf.keras.models.load_model('my_model')
+  #x_model = loaded_model = tf.keras.models.load_model('my_model')
+    #x_model = tf.keras.models.load_model('my_model.h5')
+  try:
+        x_model = tf.keras.models.load_model('my_model.h5')
+  except:
+        x_model = tf.keras.layers.TFSMLayer('my_model', call_endpoint='serving_default')
   y_pred = x_model.predict(X_test_df)
   #predicition = []
   #for i in list(y_pred):