Update README.md

README.md

@@ -12,10 +12,10 @@ from transformers import GPT2Tokenizer, GPT2LMHeadModel
 import torch
 
 
-tokenizer = GPT2Tokenizer.from_pretrained('ericksgarbi/prompt_generator')
+tokenizer = GPT2Tokenizer.from_pretrained('sgarbi/prompt_generator')
 tokenizer.pad_token = tokenizer.eos_token
 
-model = GPT2LMHeadModel.from_pretrained('ericksgarbi/prompt_generator')
+model = GPT2LMHeadModel.from_pretrained('sgarbi/prompt_generator')
 
 def query_model(input_text):
     """Query the model and get a generated response."""
@@ -35,8 +35,7 @@ def query_model(input_text):
     return decoded[tag_index + len('<|prompt|>'):]
 
 # Example use
-print(query_model("Microbiologist Researcher"))
-
+print(query_model("Space X Engineer"))
 ```
 
 ## Examples