Added requirements.txt

app.py

@@ -2,8 +2,9 @@ import numpy as np
 from scipy import signal
 import huggingface_hub  # for loading model
 import streamlit as st
-from transformers import AutoModel
-import torch
+# from transformers import AutoModel
+from transformers import TFAutoModel
+import torch  # Needed for importing torch to use in the transformers model
 # HELLO HUGGING FACE
 
 
@@ -135,9 +136,9 @@ interp_length = st.selectbox("Interpolation Length", interpolation_options)
 # 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 = AutoModel.from_pretrained("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 = AutoModel.from_pretrained("cmudrc/2d-lattice-decoder")
+decoder_model = TFAutoModel.from_pretrained("cmudrc/2d-lattice-decoder")