Spaces:

amanneo
/

poem-bot

Sleeping

amanneo commited on Oct 19, 2022

Commit

d84944c

1 Parent(s): 74be9b2

feat: Added required files

Files changed (1) hide show

app.py CHANGED Viewed

@@ -13,7 +13,7 @@ model = tf.keras.Sequential([
             tf.keras.layers.Bidirectional(tf.keras.layers.LSTM(120)),
             tf.keras.layers.Dense(vocab_size, activation='softmax')
 ])
-model.load_weights("poembot_weight/")
 word_token = json.load("word_token.json")
 # Main function
 def text_generator(seed_text,n):

             tf.keras.layers.Bidirectional(tf.keras.layers.LSTM(120)),
             tf.keras.layers.Dense(vocab_size, activation='softmax')
 ])
+model.load_weights("model_weight/")
 word_token = json.load("word_token.json")
 # Main function
 def text_generator(seed_text,n):