ggirishg commited on
Commit
0baf760
·
verified ·
1 Parent(s): 8f1708c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -21,9 +21,9 @@ try:
21
  except Exception as e:
22
  st.warning(f"Could not set GPU memory growth: {e}")
23
 
24
- model_path = 'TrillsonFeature_model'
25
- m = hub.load(model_path)
26
-
27
  class TransformerEncoder(tf.keras.layers.Layer):
28
  def __init__(self, embed_dim, num_heads, ff_dim, rate=0.01, **kwargs):
29
  super(TransformerEncoder, self).__init__(**kwargs)
 
21
  except Exception as e:
22
  st.warning(f"Could not set GPU memory growth: {e}")
23
 
24
+ # model_path = 'TrillsonFeature_model'
25
+ # m = hub.load(model_path)
26
+ m = hub.KerasLayer('https://tfhub.dev/google/nonsemantic-speech-benchmark/trillsson4/1')
27
  class TransformerEncoder(tf.keras.layers.Layer):
28
  def __init__(self, embed_dim, num_heads, ff_dim, rate=0.01, **kwargs):
29
  super(TransformerEncoder, self).__init__(**kwargs)