CustomGPT

from transformers import AutoModel
import tiktoken

# Load tokenizer
tokenizer = tiktoken.get_encoding("gpt2")

# Load model
model_id = "FarhanAK128/CustomGPT"
model = AutoModel.from_pretrained(
    model_id,
    trust_remote_code=True
)

# Example prediction
input = {'instruction': 'Rewrite the sentence using a simile.',
         'input': 'The car is very fast.'
        }

response = model.generate_response(input, tokenizer)
print(response) # The car is as fast as a cheetah.