Hugging Face's logo

Spaces:
cmudrc
/
Interp_Imaging
Sleeping

App Files Community
marta-marta commited on
Commit
36fa435
·
1 Parent(s): 29b4b22

Updating requirements

Browse files
Files changed (2) hide show
  1. app.py +5 -4
  2. requirements.txt +1 -1
app.py CHANGED
@@ -2,7 +2,7 @@ import numpy as np
2
  from scipy import signal
3
  import huggingface_hub # for loading model
4
  import streamlit as st
5
- import tensorflow
6
  # HELLO HUGGING FACE
7
 
8
 
@@ -133,9 +133,10 @@ interp_length = st.selectbox("Interpolation Length", interpolation_options)
133
 
134
  # Load the models from existing huggingface model
135
  # Load the encoder model
136
- encoder_model_boxes = huggingface_hub.from_pretrained_keras("cmudrc/2d-lattice-encoder")
 
137
  # Load the decoder model
138
- decoder_model_boxes = huggingface_hub.from_pretrained_keras("cmudrc/2d-lattice-decoder")
139
-
140
 
141
 
 
2
  from scipy import signal
3
  import huggingface_hub # for loading model
4
  import streamlit as st
5
+ from transformers import AutoModel
6
  # HELLO HUGGING FACE
7
 
8
 
 
133
 
134
  # Load the models from existing huggingface model
135
  # Load the encoder model
136
+ # encoder_model_boxes = huggingface_hub.from_pretrained_keras("cmudrc/2d-lattice-encoder")
137
+ encoder_model = AutoModel.from_pretrained("cmudrc/2d-lattice-encoder")
138
  # Load the decoder model
139
+ # decoder_model_boxes = huggingface_hub.from_pretrained_keras("cmudrc/2d-lattice-decoder")
140
+ decoder_model = AutoModel.from_pretrained("cmudrc/2d-lattice-decoder")
141
 
142
 
requirements.txt CHANGED
@@ -2,4 +2,4 @@ huggingface_hub==0.12.0
2
  numpy==1.21.5
3
  scipy==1.9.1
4
  streamlit==1.18.1
5
- tensorflow==2.10.0
 
2
  numpy==1.21.5
3
  scipy==1.9.1
4
  streamlit==1.18.1
5
+ transformers==4.26.0