Updating to get the model imports to work

app.py

@@ -281,10 +281,10 @@ if st.button("Generate Endpoint Images"):
 
 # Load the models from existing huggingface model
 # Load the encoder model
-# encoder_model_boxes = huggingface_hub.from_pretrained_keras("cmudrc/2d-lattice-encoder")
-encoder_model = TFAutoModel.from_pretrained("cmudrc/2d-lattice-encoder")
+encoder_model_boxes = huggingface_hub.from_pretrained_keras("cmudrc/2d-lattice-encoder")
+# encoder_model = TFAutoModel.from_pretrained("cmudrc/2d-lattice-encoder")
 # Load the decoder model
-# decoder_model_boxes = huggingface_hub.from_pretrained_keras("cmudrc/2d-lattice-decoder")
-decoder_model = TFAutoModel.from_pretrained("cmudrc/2d-lattice-decoder")
+decoder_model_boxes = huggingface_hub.from_pretrained_keras("cmudrc/2d-lattice-decoder")
+# decoder_model = TFAutoModel.from_pretrained("cmudrc/2d-lattice-decoder")