sysmlAI / README.md
MohamedIFQ's picture
Update README.md
740bf29 verified
|
raw
history blame
489 Bytes
metadata
license: mit

tags: - text-generation

Model Name

This is a fine-tuned model for text generation.

Usage

from transformers import AutoModelForCausalLM, AutoTokenizer, pipeline

model_name = "MohamedIFQ/sysmlAI"
tokenizer = AutoTokenizer.from_pretrained(model_name)
model = AutoModelForCausalLM.from_pretrained(model_name)

nlp = pipeline("text-generation", model=model, tokenizer=tokenizer)

prompt = "Your prompt here"
result = nlp(prompt)
print(result)