Update app.py

app.py

@@ -71,7 +71,7 @@ class TokenAndPositionEmbedding(tf.keras.layers.Layer):
         })
         return config
 
-def load_model(filename="zero_30m"):
+def load_model(filename="tg_small"):
     model = tf.keras.models.load_model(f'{filename}.h5', custom_objects={
         'TokenAndPositionEmbedding': TokenAndPositionEmbedding,
         'TransformerBlock': TransformerBlock