Spaces:
Sleeping
Sleeping
AILaborant
commited on
Update app.py
Browse files
app.py
CHANGED
@@ -71,7 +71,7 @@ class TokenAndPositionEmbedding(tf.keras.layers.Layer):
|
|
71 |
})
|
72 |
return config
|
73 |
|
74 |
-
def load_model(filename="
|
75 |
model = tf.keras.models.load_model(f'{filename}.h5', custom_objects={
|
76 |
'TokenAndPositionEmbedding': TokenAndPositionEmbedding,
|
77 |
'TransformerBlock': TransformerBlock
|
|
|
71 |
})
|
72 |
return config
|
73 |
|
74 |
+
def load_model(filename="tg-small"):
|
75 |
model = tf.keras.models.load_model(f'{filename}.h5', custom_objects={
|
76 |
'TokenAndPositionEmbedding': TokenAndPositionEmbedding,
|
77 |
'TransformerBlock': TransformerBlock
|