Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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)
|