Skip to content

Commit

Permalink
Update create_model.py
Browse files Browse the repository at this point in the history
  • Loading branch information
nakira974 committed Apr 20, 2023
1 parent 7cc3716 commit 88795df
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion descrivizio001/create_model.py
Original file line number Diff line number Diff line change
Expand Up @@ -148,7 +148,7 @@ def prepare_data_for_text_generation(embedding_matrix, dataframe):
text_generation_model.fit([decoder_input_data], [decoder_target_data], epochs=10, batch_size=32)
output = text_generation_model.predict([encoder_input_data])

login(token="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")
login(token="")
tokenizer = AutoTokenizer.from_pretrained("bert-base-uncased")
decoded_output = tokenizer.decode(output[0], skip_special_tokens=True)
print(decoded_output)
Expand Down

0 comments on commit 88795df

Please sign in to comment.