AILaborant commited on
Commit
ac82d79
·
verified ·
1 Parent(s): 02b9cf0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -71,7 +71,7 @@ class TokenAndPositionEmbedding(tf.keras.layers.Layer):
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
 
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