check loading of UniRep parameters

app.py

@@ -89,10 +89,9 @@ def display_dti():
                 encoder = SeqVecEmbedder()
                 embedding = encoder([sequence])
                 embedding = encoder.reduce_per_protein(embedding)
-                st.write(f'SeqVec embedding: {embedding}')
             elif selected_encoder == 'UniRep':
-                #from jax_unirep.utils import load_params
-                #params = load_params() 
+                from jax_unirep.utils import load_params
+                params = load_params() 
                 from jax_unirep.featurize import get_reps                               
                 embedding, h_final, c_final = get_reps([sequence])
                 embedding = embedding.mean(axis=0)
@@ -108,6 +107,9 @@ def display_dti():
                 embedding = encoder.reduce_per_protein(embedding)
             else: 
                 st.write('No pre-trained version of HyperPCM is available for the chosen encoder.')
+                embeddning = None
+            if embedding:
+                st.write(f'{selected_encoder} embedding: {embedding}')
 
 
 def display_protein():