Updating to get the model imports to work

app.py

@@ -2,6 +2,7 @@ import numpy as np
 from scipy import signal
 import math
 # import huggingface_hub  # for loading model
+from huggingface_hub import from_pretrained_keras
 import streamlit as st
 # from transformers import AutoModel
 # from transformers import TFAutoModel
@@ -281,10 +282,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_boxes = 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_boxes = from_pretrained_keras("cmudrc/2d-lattice-decoder")
 # decoder_model = TFAutoModel.from_pretrained("cmudrc/2d-lattice-decoder")
 
 

requirements.txt

@@ -1,3 +1,4 @@
+huggingface_hub==0.12.0
 matplotlib==3.5.2
 numpy==1.21.5
 scipy==1.9.1