Views
No views yet
1from transformers import AutoModelForCausalLM, AutoTokenizer
2
3# Load the model
4model = AutoModelForCausalLM.from_pretrained("gdshaji/mistral7B-31-10-24")
5tokenizer = AutoTokenizer.from_pretrained("gdshaji/mistral7B-31-10-24")
6
7# Example usage
8text = "Your input text here"
9inputs = tokenizer(text, return_tensors="pt")
10outputs = model.generate(**inputs)
11result = tokenizer.decode(outputs[0], skip_special_tokens=True)