Update README.md

README.md

@@ -25,11 +25,11 @@ from transformers import AutoModelForCausalLM, AutoTokenizer
 import torch
 model_path = "janbol/pythia-reverse-160m-Flan"
 
-#Load the tokenizer and model
+# Load the tokenizer and model
 tokenizer = AutoTokenizer.from_pretrained(model_path)
 model = AutoModelForCausalLM.from_pretrained(model_path)
 
-#Move model to GPU if available
+# Move model to GPU if available
 device = "cuda" if torch.cuda.is_available() else "cpu"
 model = model.to(device)
 
@@ -59,7 +59,8 @@ def generate_flipped(prompt, max_length=100):
 
     # Flip back and decode
     return tokenizer.decode(torch.flip(output, (1,))[0], skip_special_tokens=True)
-print(generate_flipped("mars is habitable planet"))
+print(generate_flipped("Thus this planet is theoretically habitable"))
+
 
 
 ####### Output ##########