Update README.md

README.md

@@ -52,8 +52,8 @@ Using `AutoModelForCausalLM` and `AutoTokenizer`
 from transformers import AutoTokenizer, AutoModelForCausalLM
 
 # Load the model and tokenizer
-tokenizer = AutoTokenizer.from_pretrained("premai-io/prem-1B-chat")
-model = AutoModelForCausalLM.from_pretrained('premai-io/prem-1B-chat', torch_dtype=torch.bfloat16)
+tokenizer = AutoTokenizer.from_pretrained("prem-research/prem-1B-chat")
+model = AutoModelForCausalLM.from_pretrained('prem-research/prem-1B-chat', torch_dtype=torch.bfloat16)
 model = model.to('cuda')
 
 # Setup terminators
@@ -88,7 +88,7 @@ import torch
 from transformers import pipeline
 
 # Load the pipeline
-pipe = pipeline("text-generation", model="premai-io/prem-1B-chat", torch_dtype=torch.bfloat16, device=0)
+pipe = pipeline("text-generation", model="prem-research/prem-1B-chat", torch_dtype=torch.bfloat16, device=0)
 
 # Prepare prompt
 messages = [