File size: 353 Bytes
05c431f |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 |
# Load the saved model
with open("model.pkl", "rb") as f:
loaded_learn = pickle.load(f)
# Define input text
TEXT = "I liked this movie because"
N_WORDS = 40
N_SENTENCES = 2
# Generate predictions
preds = [loaded_learn.predict(TEXT, N_WORDS, temperature=0.75) for _ in range(N_SENTENCES)]
# Print predictions
print("\n".join(preds))
|